pose_list.py

import json
import os
import activities
import argparse

# Collects all the json files from the folder and assign a
# unique ID for each one
def generate_activity_list(json_dir):
    files = os.listdir(json_dir)

    activities_list = []

    for f in files:
        with open(json_dir + f) as file:
            Json_dict = json.load(file)

            for video in list(Json_dict.keys()):
                for activity in list(Json_dict[video]):
                    if (activity['label'] not in activities_list):
                        activities_list.append(activity['label'])

    activities_ids = dict(map(reversed, enumerate(activities_list)))

    return activities_ids

# Show the frequency of each activity
def show_activities_frequency(json_dir):
    files = os.listdir(json_dir)

    activities_dic = {}

    activities_list = activities.activities

    for f in files:
        with open(json_dir + f) as file:
            Json_dict = json.load(file)

            for video in list(Json_dict.keys()):
                for activity in list(Json_dict[video]):
                    if activity['label'] in activities_list:
                        if (activity['label'] not in activities_dic):
                            activities_dic[activity['label']] = 0
                        else:
                            activities_dic[activity['label']] += 1

    return activities_dic

# Main function
def main(json):

    print('\nTotal list of activities\n')
    activities = generate_activity_list(json)
    for i in activities:
        print(i)

    print('\nFrequency of each activity chosen to training\n')
    activities = show_activities_frequency(json)
    for i in activities:
        print(i, ': ', activities[i])

if __name__ == '__main__':

    parser = argparse.ArgumentParser(description='Print ')
    parser.add_argument('--json', dest='json', type=str, default='json/', help='path of the json files')
    args = parser.parse_args()

    main(args.json)