From 9c91e4bafc811e54a142cfd4bcf5941a870531f1 Mon Sep 17 00:00:00 2001 From: xt <2761564455@qq.com> Date: Thu, 14 Jan 2021 10:49:10 +0800 Subject: [PATCH 1/7] del paddlevideo_loc,mv paddlevieo_clas to tools --- MANIFEST.in | 7 ++----- __init__.py | 2 +- paddlevideo_loc.py | 0 tools/__init__.py | 3 ++- paddlevideo_clas.py => tools/paddlevideo_clas.py | 8 +++----- 5 files changed, 8 insertions(+), 12 deletions(-) delete mode 100644 paddlevideo_loc.py rename paddlevideo_clas.py => tools/paddlevideo_clas.py (98%) diff --git a/MANIFEST.in b/MANIFEST.in index 21ef7f793..cf4711257 100644 --- a/MANIFEST.in +++ b/MANIFEST.in @@ -2,9 +2,6 @@ include LICENSE include README.md include tools/__init__.py include tools/utils.py +include tools/paddlevideo_clas.py -recursive-include paddlevideo/ *.py *.txt - - - - +recursive-include paddlevideo/ *.py *.txt \ No newline at end of file diff --git a/__init__.py b/__init__.py index c507311cc..c4f28815b 100644 --- a/__init__.py +++ b/__init__.py @@ -13,4 +13,4 @@ # limitations under the License. __all__ = ['PaddleVideo'] -from .paddlevideo_clas import PaddleVideo \ No newline at end of file +from .tools import PaddleVideo \ No newline at end of file diff --git a/paddlevideo_loc.py b/paddlevideo_loc.py deleted file mode 100644 index e69de29bb..000000000 diff --git a/tools/__init__.py b/tools/__init__.py index d332d628e..c14145730 100644 --- a/tools/__init__.py +++ b/tools/__init__.py @@ -12,4 +12,5 @@ # See the License for the specific language governing permissions and # limitations under the License. -from . import utils \ No newline at end of file +from . import utils +from .paddlevideo_clas import PaddleVideo \ No newline at end of file diff --git a/paddlevideo_clas.py b/tools/paddlevideo_clas.py similarity index 98% rename from paddlevideo_clas.py rename to tools/paddlevideo_clas.py index 7543f52e8..3f3482746 100644 --- a/paddlevideo_clas.py +++ b/tools/paddlevideo_clas.py @@ -48,7 +48,7 @@ BASE_INFERENCE_MODEL_DIR = os.path.join(BASE_DIR, 'inference_model') BASE_VIDEOS_DIR = os.path.join(BASE_DIR, 'videos') -model_names = {'TSM','TSN','PPTSM'} +model_names = {'ppTSM','TSM','TSN'} def create_paddle_predictor(args): @@ -78,7 +78,6 @@ def create_paddle_predictor(args): return predictor - def download_with_progressbar(url, save_path): response = requests.get(url, stream=True) total_size_in_bytes = int(response.headers.get('content-length', 0)) @@ -95,7 +94,7 @@ def download_with_progressbar(url, save_path): def maybe_download(model_storage_directory, url): # using custom model tar_file_name_list = [ - 'inference.pdiparams', 'inference.pdiparams.info', 'inference.pdmodel' #pdiparams,和pdmodel直接下载 + 'inference.pdiparams', 'inference.pdiparams.info', 'inference.pdmodel' ] if not os.path.exists( os.path.join(model_storage_directory, 'inference.pdiparams') @@ -225,8 +224,7 @@ def __init__(self, **kwargs): raise Exception( 'Please input model name that you want to use!') if process_params.model_name in model_names: - url = 'https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/inference/{}_infer.tar'.format( - process_params.model_name) + url = 'https://videotag.bj.bcebos.com/PaddleVideo/InferenceModel/{}_infer.tar'.format(process_params.model_name) if not os.path.exists( os.path.join(BASE_INFERENCE_MODEL_DIR, process_params.model_name)): From 918bddefce258a0a8c4d4bb4001b94a7b742f261 Mon Sep 17 00:00:00 2001 From: xt <2761564455@qq.com> Date: Thu, 14 Jan 2021 11:22:46 +0800 Subject: [PATCH 2/7] update --- paddlevideo/utils/Kinetics-400_label_list.txt | 1 - requirements.txt | 2 +- tools/paddlevideo_clas.py | 24 ++----------------- 3 files changed, 3 insertions(+), 24 deletions(-) delete mode 100644 paddlevideo/utils/Kinetics-400_label_list.txt diff --git a/paddlevideo/utils/Kinetics-400_label_list.txt b/paddlevideo/utils/Kinetics-400_label_list.txt deleted file mode 100644 index 3a739b37e..000000000 --- a/paddlevideo/utils/Kinetics-400_label_list.txt +++ /dev/null @@ -1 +0,0 @@ -{"abseiling": "0", "air_drumming": "1", "answering_questions": "2", "applauding": "3", "applying_cream": "4", "archery": "5", "arm_wrestling": "6", "arranging_flowers": "7", "assembling_computer": "8", "auctioning": "9", "baby_waking_up": "10", "baking_cookies": "11", "balloon_blowing": "12", "bandaging": "13", "barbequing": "14", "bartending": "15", "beatboxing": "16", "bee_keeping": "17", "belly_dancing": "18", "bench_pressing": "19", "bending_back": "20", "bending_metal": "21", "biking_through_snow": "22", "blasting_sand": "23", "blowing_glass": "24", "blowing_leaves": "25", "blowing_nose": "26", "blowing_out_candles": "27", "bobsledding": "28", "bookbinding": "29", "bouncing_on_trampoline": "30", "bowling": "31", "braiding_hair": "32", "breading_or_breadcrumbing": "33", "breakdancing": "34", "brush_painting": "35", "brushing_hair": "36", "brushing_teeth": "37", "building_cabinet": "38", "building_shed": "39", "bungee_jumping": "40", "busking": "41", "canoeing_or_kayaking": "42", "capoeira": "43", "carrying_baby": "44", "cartwheeling": "45", "carving_pumpkin": "46", "catching_fish": "47", "catching_or_throwing_baseball": "48", "catching_or_throwing_frisbee": "49", "catching_or_throwing_softball": "50", "celebrating": "51", "changing_oil": "52", "changing_wheel": "53", "checking_tires": "54", "cheerleading": "55", "chopping_wood": "56", "clapping": "57", "clay_pottery_making": "58", "clean_and_jerk": "59", "cleaning_floor": "60", "cleaning_gutters": "61", "cleaning_pool": "62", "cleaning_shoes": "63", "cleaning_toilet": "64", "cleaning_windows": "65", "climbing_a_rope": "66", "climbing_ladder": "67", "climbing_tree": "68", "contact_juggling": "69", "cooking_chicken": "70", "cooking_egg": "71", "cooking_on_campfire": "72", "cooking_sausages": "73", "counting_money": "74", "country_line_dancing": "75", "cracking_neck": "76", "crawling_baby": "77", "crossing_river": "78", "crying": "79", "curling_hair": "80", "cutting_nails": "81", "cutting_pineapple": "82", "cutting_watermelon": "83", "dancing_ballet": "84", "dancing_charleston": "85", "dancing_gangnam_style": "86", "dancing_macarena": "87", "deadlifting": "88", "decorating_the_christmas_tree": "89", "digging": "90", "dining": "91", "disc_golfing": "92", "diving_cliff": "93", "dodgeball": "94", "doing_aerobics": "95", "doing_laundry": "96", "doing_nails": "97", "drawing": "98", "dribbling_basketball": "99", "drinking": "100", "drinking_beer": "101", "drinking_shots": "102", "driving_car": "103", "driving_tractor": "104", "drop_kicking": "105", "drumming_fingers": "106", "dunking_basketball": "107", "dying_hair": "108", "eating_burger": "109", "eating_cake": "110", "eating_carrots": "111", "eating_chips": "112", "eating_doughnuts": "113", "eating_hotdog": "114", "eating_ice_cream": "115", "eating_spaghetti": "116", "eating_watermelon": "117", "egg_hunting": "118", "exercising_arm": "119", "exercising_with_an_exercise_ball": "120", "extinguishing_fire": "121", "faceplanting": "122", "feeding_birds": "123", "feeding_fish": "124", "feeding_goats": "125", "filling_eyebrows": "126", "finger_snapping": "127", "fixing_hair": "128", "flipping_pancake": "129", "flying_kite": "130", "folding_clothes": "131", "folding_napkins": "132", "folding_paper": "133", "front_raises": "134", "frying_vegetables": "135", "garbage_collecting": "136", "gargling": "137", "getting_a_haircut": "138", "getting_a_tattoo": "139", "giving_or_receiving_award": "140", "golf_chipping": "141", "golf_driving": "142", "golf_putting": "143", "grinding_meat": "144", "grooming_dog": "145", "grooming_horse": "146", "gymnastics_tumbling": "147", "hammer_throw": "148", "headbanging": "149", "headbutting": "150", "high_jump": "151", "high_kick": "152", "hitting_baseball": "153", "hockey_stop": "154", "holding_snake": "155", "hopscotch": "156", "hoverboarding": "157", "hugging": "158", "hula_hooping": "159", "hurdling": "160", "hurling_(sport)": "161", "ice_climbing": "162", "ice_fishing": "163", "ice_skating": "164", "ironing": "165", "javelin_throw": "166", "jetskiing": "167", "jogging": "168", "juggling_balls": "169", "juggling_fire": "170", "juggling_soccer_ball": "171", "jumping_into_pool": "172", "jumpstyle_dancing": "173", "kicking_field_goal": "174", "kicking_soccer_ball": "175", "kissing": "176", "kitesurfing": "177", "knitting": "178", "krumping": "179", "laughing": "180", "laying_bricks": "181", "long_jump": "182", "lunge": "183", "making_a_cake": "184", "making_a_sandwich": "185", "making_bed": "186", "making_jewelry": "187", "making_pizza": "188", "making_snowman": "189", "making_sushi": "190", "making_tea": "191", "marching": "192", "massaging_back": "193", "massaging_feet": "194", "massaging_legs": "195", "massaging_person's_head": "196", "milking_cow": "197", "mopping_floor": "198", "motorcycling": "199", "moving_furniture": "200", "mowing_lawn": "201", "news_anchoring": "202", "opening_bottle": "203", "opening_present": "204", "paragliding": "205", "parasailing": "206", "parkour": "207", "passing_American_football_(in_game)": "208", "passing_American_football_(not_in_game)": "209", "peeling_apples": "210", "peeling_potatoes": "211", "petting_animal_(not_cat)": "212", "petting_cat": "213", "picking_fruit": "214", "planting_trees": "215", "plastering": "216", "playing_accordion": "217", "playing_badminton": "218", "playing_bagpipes": "219", "playing_basketball": "220", "playing_bass_guitar": "221", "playing_cards": "222", "playing_cello": "223", "playing_chess": "224", "playing_clarinet": "225", "playing_controller": "226", "playing_cricket": "227", "playing_cymbals": "228", "playing_didgeridoo": "229", "playing_drums": "230", "playing_flute": "231", "playing_guitar": "232", "playing_harmonica": "233", "playing_harp": "234", "playing_ice_hockey": "235", "playing_keyboard": "236", "playing_kickball": "237", "playing_monopoly": "238", "playing_organ": "239", "playing_paintball": "240", "playing_piano": "241", "playing_poker": "242", "playing_recorder": "243", "playing_saxophone": "244", "playing_squash_or_racquetball": "245", "playing_tennis": "246", "playing_trombone": "247", "playing_trumpet": "248", "playing_ukulele": "249", "playing_violin": "250", "playing_volleyball": "251", "playing_xylophone": "252", "pole_vault": "253", "presenting_weather_forecast": "254", "pull_ups": "255", "pumping_fist": "256", "pumping_gas": "257", "punching_bag": "258", "punching_person_(boxing)": "259", "push_up": "260", "pushing_car": "261", "pushing_cart": "262", "pushing_wheelchair": "263", "reading_book": "264", "reading_newspaper": "265", "recording_music": "266", "riding_a_bike": "267", "riding_camel": "268", "riding_elephant": "269", "riding_mechanical_bull": "270", "riding_mountain_bike": "271", "riding_mule": "272", "riding_or_walking_with_horse": "273", "riding_scooter": "274", "riding_unicycle": "275", "ripping_paper": "276", "robot_dancing": "277", "rock_climbing": "278", "rock_scissors_paper": "279", "roller_skating": "280", "running_on_treadmill": "281", "sailing": "282", "salsa_dancing": "283", "sanding_floor": "284", "scrambling_eggs": "285", "scuba_diving": "286", "setting_table": "287", "shaking_hands": "288", "shaking_head": "289", "sharpening_knives": "290", "sharpening_pencil": "291", "shaving_head": "292", "shaving_legs": "293", "shearing_sheep": "294", "shining_shoes": "295", "shooting_basketball": "296", "shooting_goal_(soccer)": "297", "shot_put": "298", "shoveling_snow": "299", "shredding_paper": "300", "shuffling_cards": "301", "side_kick": "302", "sign_language_interpreting": "303", "singing": "304", "situp": "305", "skateboarding": "306", "ski_jumping": "307", "skiing_(not_slalom_or_crosscountry)": "308", "skiing_crosscountry": "309", "skiing_slalom": "310", "skipping_rope": "311", "skydiving": "312", "slacklining": "313", "slapping": "314", "sled_dog_racing": "315", "smoking": "316", "smoking_hookah": "317", "snatch_weight_lifting": "318", "sneezing": "319", "sniffing": "320", "snorkeling": "321", "snowboarding": "322", "snowkiting": "323", "snowmobiling": "324", "somersaulting": "325", "spinning_poi": "326", "spray_painting": "327", "spraying": "328", "springboard_diving": "329", "squat": "330", "sticking_tongue_out": "331", "stomping_grapes": "332", "stretching_arm": "333", "stretching_leg": "334", "strumming_guitar": "335", "surfing_crowd": "336", "surfing_water": "337", "sweeping_floor": "338", "swimming_backstroke": "339", "swimming_breast_stroke": "340", "swimming_butterfly_stroke": "341", "swing_dancing": "342", "swinging_legs": "343", "swinging_on_something": "344", "sword_fighting": "345", "tai_chi": "346", "taking_a_shower": "347", "tango_dancing": "348", "tap_dancing": "349", "tapping_guitar": "350", "tapping_pen": "351", "tasting_beer": "352", "tasting_food": "353", "testifying": "354", "texting": "355", "throwing_axe": "356", "throwing_ball": "357", "throwing_discus": "358", "tickling": "359", "tobogganing": "360", "tossing_coin": "361", "tossing_salad": "362", "training_dog": "363", "trapezing": "364", "trimming_or_shaving_beard": "365", "trimming_trees": "366", "triple_jump": "367", "tying_bow_tie": "368", "tying_knot_(not_on_a_tie)": "369", "tying_tie": "370", "unboxing": "371", "unloading_truck": "372", "using_computer": "373", "using_remote_controller_(not_gaming)": "374", "using_segway": "375", "vault": "376", "waiting_in_line": "377", "walking_the_dog": "378", "washing_dishes": "379", "washing_feet": "380", "washing_hair": "381", "washing_hands": "382", "water_skiing": "383", "water_sliding": "384", "watering_plants": "385", "waxing_back": "386", "waxing_chest": "387", "waxing_eyebrows": "388", "waxing_legs": "389", "weaving_basket": "390", "welding": "391", "whistling": "392", "windsurfing": "393", "wrapping_present": "394", "wrestling": "395", "writing": "396", "yawning": "397", "yoga": "398", "zumba": "399"} \ No newline at end of file diff --git a/requirements.txt b/requirements.txt index b91895ede..3a1932986 100644 --- a/requirements.txt +++ b/requirements.txt @@ -4,4 +4,4 @@ PyYAML>=5.1 numpy decord pandas -opencv-python>=4.2.0.32 +opencv-python>=4.1.1 diff --git a/tools/paddlevideo_clas.py b/tools/paddlevideo_clas.py index 3f3482746..fe2d53b52 100644 --- a/tools/paddlevideo_clas.py +++ b/tools/paddlevideo_clas.py @@ -240,14 +240,14 @@ def __init__(self, **kwargs): process_params.params_file = os.path.join( download_path, 'inference.pdiparams') process_params.label_name_path = os.path.join( - __dir__, 'paddlevideo/utils/Kinetics-400_label_list.txt') + __dir__, 'Kinetics-400_label_list.txt') else: raise Exception( 'If you want to use your own model, Please input model_file as model path!' ) else: print('Using user-specified model and params!') - print("process params are as follows: \n{}".format(process_params))#一个字典 + print("process params are as follows: \n{}".format(process_params)) self.label_name_dict = load_label_name_dict( process_params.label_name_path) @@ -319,26 +319,6 @@ def predict(self,video): total_result.append(result) return total_result - # # for PaddleHubServing - # if self.args.hubserving: - # v = self.args.video_file - # # for predict only - # else: - # v = utils.decode(video, self.args) - # assert v is not None, "Error in loading video: {}".format( - # self.args.video_file) - # inputs = utils.preprocess(v, self.args) - # inputs = np.expand_dims( - # inputs, axis=0).repeat( - # self.args.batch_size, axis=0).copy() - # - # input_tensor.copy_from_cpu(inputs) - # - # self.predictor.run() - # - # output = output_tensor.copy_to_cpu() - # return utils.postprocess(output, self.args) - def main(): # for cmd args = parse_args(mMain=True) From fbab977cf1ac76a0da54d476eceeee79fafc6fb8 Mon Sep 17 00:00:00 2001 From: xt <2761564455@qq.com> Date: Thu, 14 Jan 2021 11:25:44 +0800 Subject: [PATCH 3/7] update K400-dict --- tools/Kinetics-400_label_list.txt | 1 + 1 file changed, 1 insertion(+) create mode 100644 tools/Kinetics-400_label_list.txt diff --git a/tools/Kinetics-400_label_list.txt b/tools/Kinetics-400_label_list.txt new file mode 100644 index 000000000..dc3e21735 --- /dev/null +++ b/tools/Kinetics-400_label_list.txt @@ -0,0 +1 @@ +{"0": "abseiling", "1": "air_drumming", "2": "answering_questions", "3": "applauding", "4": "applying_cream", "5": "archery", "6": "arm_wrestling", "7": "arranging_flowers", "8": "assembling_computer", "9": "auctioning", "10": "baby_waking_up", "11": "baking_cookies", "12": "balloon_blowing", "13": "bandaging", "14": "barbequing", "15": "bartending", "16": "beatboxing", "17": "bee_keeping", "18": "belly_dancing", "19": "bench_pressing", "20": "bending_back", "21": "bending_metal", "22": "biking_through_snow", "23": "blasting_sand", "24": "blowing_glass", "25": "blowing_leaves", "26": "blowing_nose", "27": "blowing_out_candles", "28": "bobsledding", "29": "bookbinding", "30": "bouncing_on_trampoline", "31": "bowling", "32": "braiding_hair", "33": "breading_or_breadcrumbing", "34": "breakdancing", "35": "brush_painting", "36": "brushing_hair", "37": "brushing_teeth", "38": "building_cabinet", "39": "building_shed", "40": "bungee_jumping", "41": "busking", "42": "canoeing_or_kayaking", "43": "capoeira", "44": "carrying_baby", "45": "cartwheeling", "46": "carving_pumpkin", "47": "catching_fish", "48": "catching_or_throwing_baseball", "49": "catching_or_throwing_frisbee", "50": "catching_or_throwing_softball", "51": "celebrating", "52": "changing_oil", "53": "changing_wheel", "54": "checking_tires", "55": "cheerleading", "56": "chopping_wood", "57": "clapping", "58": "clay_pottery_making", "59": "clean_and_jerk", "60": "cleaning_floor", "61": "cleaning_gutters", "62": "cleaning_pool", "63": "cleaning_shoes", "64": "cleaning_toilet", "65": "cleaning_windows", "66": "climbing_a_rope", "67": "climbing_ladder", "68": "climbing_tree", "69": "contact_juggling", "70": "cooking_chicken", "71": "cooking_egg", "72": "cooking_on_campfire", "73": "cooking_sausages", "74": "counting_money", "75": "country_line_dancing", "76": "cracking_neck", "77": "crawling_baby", "78": "crossing_river", "79": "crying", "80": "curling_hair", "81": "cutting_nails", "82": "cutting_pineapple", "83": "cutting_watermelon", "84": "dancing_ballet", "85": "dancing_charleston", "86": "dancing_gangnam_style", "87": "dancing_macarena", "88": "deadlifting", "89": "decorating_the_christmas_tree", "90": "digging", "91": "dining", "92": "disc_golfing", "93": "diving_cliff", "94": "dodgeball", "95": "doing_aerobics", "96": "doing_laundry", "97": "doing_nails", "98": "drawing", "99": "dribbling_basketball", "100": "drinking", "101": "drinking_beer", "102": "drinking_shots", "103": "driving_car", "104": "driving_tractor", "105": "drop_kicking", "106": "drumming_fingers", "107": "dunking_basketball", "108": "dying_hair", "109": "eating_burger", "110": "eating_cake", "111": "eating_carrots", "112": "eating_chips", "113": "eating_doughnuts", "114": "eating_hotdog", "115": "eating_ice_cream", "116": "eating_spaghetti", "117": "eating_watermelon", "118": "egg_hunting", "119": "exercising_arm", "120": "exercising_with_an_exercise_ball", "121": "extinguishing_fire", "122": "faceplanting", "123": "feeding_birds", "124": "feeding_fish", "125": "feeding_goats", "126": "filling_eyebrows", "127": "finger_snapping", "128": "fixing_hair", "129": "flipping_pancake", "130": "flying_kite", "131": "folding_clothes", "132": "folding_napkins", "133": "folding_paper", "134": "front_raises", "135": "frying_vegetables", "136": "garbage_collecting", "137": "gargling", "138": "getting_a_haircut", "139": "getting_a_tattoo", "140": "giving_or_receiving_award", "141": "golf_chipping", "142": "golf_driving", "143": "golf_putting", "144": "grinding_meat", "145": "grooming_dog", "146": "grooming_horse", "147": "gymnastics_tumbling", "148": "hammer_throw", "149": "headbanging", "150": "headbutting", "151": "high_jump", "152": "high_kick", "153": "hitting_baseball", "154": "hockey_stop", "155": "holding_snake", "156": "hopscotch", "157": "hoverboarding", "158": "hugging", "159": "hula_hooping", "160": "hurdling", "161": "hurling_(sport)", "162": "ice_climbing", "163": "ice_fishing", "164": "ice_skating", "165": "ironing", "166": "javelin_throw", "167": "jetskiing", "168": "jogging", "169": "juggling_balls", "170": "juggling_fire", "171": "juggling_soccer_ball", "172": "jumping_into_pool", "173": "jumpstyle_dancing", "174": "kicking_field_goal", "175": "kicking_soccer_ball", "176": "kissing", "177": "kitesurfing", "178": "knitting", "179": "krumping", "180": "laughing", "181": "laying_bricks", "182": "long_jump", "183": "lunge", "184": "making_a_cake", "185": "making_a_sandwich", "186": "making_bed", "187": "making_jewelry", "188": "making_pizza", "189": "making_snowman", "190": "making_sushi", "191": "making_tea", "192": "marching", "193": "massaging_back", "194": "massaging_feet", "195": "massaging_legs", "196": "massaging_person's_head", "197": "milking_cow", "198": "mopping_floor", "199": "motorcycling", "200": "moving_furniture", "201": "mowing_lawn", "202": "news_anchoring", "203": "opening_bottle", "204": "opening_present", "205": "paragliding", "206": "parasailing", "207": "parkour", "208": "passing_American_football_(in_game)", "209": "passing_American_football_(not_in_game)", "210": "peeling_apples", "211": "peeling_potatoes", "212": "petting_animal_(not_cat)", "213": "petting_cat", "214": "picking_fruit", "215": "planting_trees", "216": "plastering", "217": "playing_accordion", "218": "playing_badminton", "219": "playing_bagpipes", "220": "playing_basketball", "221": "playing_bass_guitar", "222": "playing_cards", "223": "playing_cello", "224": "playing_chess", "225": "playing_clarinet", "226": "playing_controller", "227": "playing_cricket", "228": "playing_cymbals", "229": "playing_didgeridoo", "230": "playing_drums", "231": "playing_flute", "232": "playing_guitar", "233": "playing_harmonica", "234": "playing_harp", "235": "playing_ice_hockey", "236": "playing_keyboard", "237": "playing_kickball", "238": "playing_monopoly", "239": "playing_organ", "240": "playing_paintball", "241": "playing_piano", "242": "playing_poker", "243": "playing_recorder", "244": "playing_saxophone", "245": "playing_squash_or_racquetball", "246": "playing_tennis", "247": "playing_trombone", "248": "playing_trumpet", "249": "playing_ukulele", "250": "playing_violin", "251": "playing_volleyball", "252": "playing_xylophone", "253": "pole_vault", "254": "presenting_weather_forecast", "255": "pull_ups", "256": "pumping_fist", "257": "pumping_gas", "258": "punching_bag", "259": "punching_person_(boxing)", "260": "push_up", "261": "pushing_car", "262": "pushing_cart", "263": "pushing_wheelchair", "264": "reading_book", "265": "reading_newspaper", "266": "recording_music", "267": "riding_a_bike", "268": "riding_camel", "269": "riding_elephant", "270": "riding_mechanical_bull", "271": "riding_mountain_bike", "272": "riding_mule", "273": "riding_or_walking_with_horse", "274": "riding_scooter", "275": "riding_unicycle", "276": "ripping_paper", "277": "robot_dancing", "278": "rock_climbing", "279": "rock_scissors_paper", "280": "roller_skating", "281": "running_on_treadmill", "282": "sailing", "283": "salsa_dancing", "284": "sanding_floor", "285": "scrambling_eggs", "286": "scuba_diving", "287": "setting_table", "288": "shaking_hands", "289": "shaking_head", "290": "sharpening_knives", "291": "sharpening_pencil", "292": "shaving_head", "293": "shaving_legs", "294": "shearing_sheep", "295": "shining_shoes", "296": "shooting_basketball", "297": "shooting_goal_(soccer)", "298": "shot_put", "299": "shoveling_snow", "300": "shredding_paper", "301": "shuffling_cards", "302": "side_kick", "303": "sign_language_interpreting", "304": "singing", "305": "situp", "306": "skateboarding", "307": "ski_jumping", "308": "skiing_(not_slalom_or_crosscountry)", "309": "skiing_crosscountry", "310": "skiing_slalom", "311": "skipping_rope", "312": "skydiving", "313": "slacklining", "314": "slapping", "315": "sled_dog_racing", "316": "smoking", "317": "smoking_hookah", "318": "snatch_weight_lifting", "319": "sneezing", "320": "sniffing", "321": "snorkeling", "322": "snowboarding", "323": "snowkiting", "324": "snowmobiling", "325": "somersaulting", "326": "spinning_poi", "327": "spray_painting", "328": "spraying", "329": "springboard_diving", "330": "squat", "331": "sticking_tongue_out", "332": "stomping_grapes", "333": "stretching_arm", "334": "stretching_leg", "335": "strumming_guitar", "336": "surfing_crowd", "337": "surfing_water", "338": "sweeping_floor", "339": "swimming_backstroke", "340": "swimming_breast_stroke", "341": "swimming_butterfly_stroke", "342": "swing_dancing", "343": "swinging_legs", "344": "swinging_on_something", "345": "sword_fighting", "346": "tai_chi", "347": "taking_a_shower", "348": "tango_dancing", "349": "tap_dancing", "350": "tapping_guitar", "351": "tapping_pen", "352": "tasting_beer", "353": "tasting_food", "354": "testifying", "355": "texting", "356": "throwing_axe", "357": "throwing_ball", "358": "throwing_discus", "359": "tickling", "360": "tobogganing", "361": "tossing_coin", "362": "tossing_salad", "363": "training_dog", "364": "trapezing", "365": "trimming_or_shaving_beard", "366": "trimming_trees", "367": "triple_jump", "368": "tying_bow_tie", "369": "tying_knot_(not_on_a_tie)", "370": "tying_tie", "371": "unboxing", "372": "unloading_truck", "373": "using_computer", "374": "using_remote_controller_(not_gaming)", "375": "using_segway", "376": "vault", "377": "waiting_in_line", "378": "walking_the_dog", "379": "washing_dishes", "380": "washing_feet", "381": "washing_hair", "382": "washing_hands", "383": "water_skiing", "384": "water_sliding", "385": "watering_plants", "386": "waxing_back", "387": "waxing_chest", "388": "waxing_eyebrows", "389": "waxing_legs", "390": "weaving_basket", "391": "welding", "392": "whistling", "393": "windsurfing", "394": "wrapping_present", "395": "wrestling", "396": "writing", "397": "yawning", "398": "yoga", "399": "zumba"} \ No newline at end of file From 43442d44cb16c93760d53cdbf1a2f3a13b085bad Mon Sep 17 00:00:00 2001 From: xt <2761564455@qq.com> Date: Thu, 14 Jan 2021 11:42:47 +0800 Subject: [PATCH 4/7] update Kinetics-400_label_list.txt --- tools/Kinetics-400_label_list.txt | 401 +++++++++++++++++++++++++++++- tools/paddlevideo_clas.py | 14 +- 2 files changed, 407 insertions(+), 8 deletions(-) diff --git a/tools/Kinetics-400_label_list.txt b/tools/Kinetics-400_label_list.txt index dc3e21735..8488908b1 100644 --- a/tools/Kinetics-400_label_list.txt +++ b/tools/Kinetics-400_label_list.txt @@ -1 +1,400 @@ -{"0": "abseiling", "1": "air_drumming", "2": "answering_questions", "3": "applauding", "4": "applying_cream", "5": "archery", "6": "arm_wrestling", "7": "arranging_flowers", "8": "assembling_computer", "9": "auctioning", "10": "baby_waking_up", "11": "baking_cookies", "12": "balloon_blowing", "13": "bandaging", "14": "barbequing", "15": "bartending", "16": "beatboxing", "17": "bee_keeping", "18": "belly_dancing", "19": "bench_pressing", "20": "bending_back", "21": "bending_metal", "22": "biking_through_snow", "23": "blasting_sand", "24": "blowing_glass", "25": "blowing_leaves", "26": "blowing_nose", "27": "blowing_out_candles", "28": "bobsledding", "29": "bookbinding", "30": "bouncing_on_trampoline", "31": "bowling", "32": "braiding_hair", "33": "breading_or_breadcrumbing", "34": "breakdancing", "35": "brush_painting", "36": "brushing_hair", "37": "brushing_teeth", "38": "building_cabinet", "39": "building_shed", "40": "bungee_jumping", "41": "busking", "42": "canoeing_or_kayaking", "43": "capoeira", "44": "carrying_baby", "45": "cartwheeling", "46": "carving_pumpkin", "47": "catching_fish", "48": "catching_or_throwing_baseball", "49": "catching_or_throwing_frisbee", "50": "catching_or_throwing_softball", "51": "celebrating", "52": "changing_oil", "53": "changing_wheel", "54": "checking_tires", "55": "cheerleading", "56": "chopping_wood", "57": "clapping", "58": "clay_pottery_making", "59": "clean_and_jerk", "60": "cleaning_floor", "61": "cleaning_gutters", "62": "cleaning_pool", "63": "cleaning_shoes", "64": "cleaning_toilet", "65": "cleaning_windows", "66": "climbing_a_rope", "67": "climbing_ladder", "68": "climbing_tree", "69": "contact_juggling", "70": "cooking_chicken", "71": "cooking_egg", "72": "cooking_on_campfire", "73": "cooking_sausages", "74": "counting_money", "75": "country_line_dancing", "76": "cracking_neck", "77": "crawling_baby", "78": "crossing_river", "79": "crying", "80": "curling_hair", "81": "cutting_nails", "82": "cutting_pineapple", "83": "cutting_watermelon", "84": "dancing_ballet", "85": "dancing_charleston", "86": "dancing_gangnam_style", "87": "dancing_macarena", "88": "deadlifting", "89": "decorating_the_christmas_tree", "90": "digging", "91": "dining", "92": "disc_golfing", "93": "diving_cliff", "94": "dodgeball", "95": "doing_aerobics", "96": "doing_laundry", "97": "doing_nails", "98": "drawing", "99": "dribbling_basketball", "100": "drinking", "101": "drinking_beer", "102": "drinking_shots", "103": "driving_car", "104": "driving_tractor", "105": "drop_kicking", "106": "drumming_fingers", "107": "dunking_basketball", "108": "dying_hair", "109": "eating_burger", "110": "eating_cake", "111": "eating_carrots", "112": "eating_chips", "113": "eating_doughnuts", "114": "eating_hotdog", "115": "eating_ice_cream", "116": "eating_spaghetti", "117": "eating_watermelon", "118": "egg_hunting", "119": "exercising_arm", "120": "exercising_with_an_exercise_ball", "121": "extinguishing_fire", "122": "faceplanting", "123": "feeding_birds", "124": "feeding_fish", "125": "feeding_goats", "126": "filling_eyebrows", "127": "finger_snapping", "128": "fixing_hair", "129": "flipping_pancake", "130": "flying_kite", "131": "folding_clothes", "132": "folding_napkins", "133": "folding_paper", "134": "front_raises", "135": "frying_vegetables", "136": "garbage_collecting", "137": "gargling", "138": "getting_a_haircut", "139": "getting_a_tattoo", "140": "giving_or_receiving_award", "141": "golf_chipping", "142": "golf_driving", "143": "golf_putting", "144": "grinding_meat", "145": "grooming_dog", "146": "grooming_horse", "147": "gymnastics_tumbling", "148": "hammer_throw", "149": "headbanging", "150": "headbutting", "151": "high_jump", "152": "high_kick", "153": "hitting_baseball", "154": "hockey_stop", "155": "holding_snake", "156": "hopscotch", "157": "hoverboarding", "158": "hugging", "159": "hula_hooping", "160": "hurdling", "161": "hurling_(sport)", "162": "ice_climbing", "163": "ice_fishing", "164": "ice_skating", "165": "ironing", "166": "javelin_throw", "167": "jetskiing", "168": "jogging", "169": "juggling_balls", "170": "juggling_fire", "171": "juggling_soccer_ball", "172": "jumping_into_pool", "173": "jumpstyle_dancing", "174": "kicking_field_goal", "175": "kicking_soccer_ball", "176": "kissing", "177": "kitesurfing", "178": "knitting", "179": "krumping", "180": "laughing", "181": "laying_bricks", "182": "long_jump", "183": "lunge", "184": "making_a_cake", "185": "making_a_sandwich", "186": "making_bed", "187": "making_jewelry", "188": "making_pizza", "189": "making_snowman", "190": "making_sushi", "191": "making_tea", "192": "marching", "193": "massaging_back", "194": "massaging_feet", "195": "massaging_legs", "196": "massaging_person's_head", "197": "milking_cow", "198": "mopping_floor", "199": "motorcycling", "200": "moving_furniture", "201": "mowing_lawn", "202": "news_anchoring", "203": "opening_bottle", "204": "opening_present", "205": "paragliding", "206": "parasailing", "207": "parkour", "208": "passing_American_football_(in_game)", "209": "passing_American_football_(not_in_game)", "210": "peeling_apples", "211": "peeling_potatoes", "212": "petting_animal_(not_cat)", "213": "petting_cat", "214": "picking_fruit", "215": "planting_trees", "216": "plastering", "217": "playing_accordion", "218": "playing_badminton", "219": "playing_bagpipes", "220": "playing_basketball", "221": "playing_bass_guitar", "222": "playing_cards", "223": "playing_cello", "224": "playing_chess", "225": "playing_clarinet", "226": "playing_controller", "227": "playing_cricket", "228": "playing_cymbals", "229": "playing_didgeridoo", "230": "playing_drums", "231": "playing_flute", "232": "playing_guitar", "233": "playing_harmonica", "234": "playing_harp", "235": "playing_ice_hockey", "236": "playing_keyboard", "237": "playing_kickball", "238": "playing_monopoly", "239": "playing_organ", "240": "playing_paintball", "241": "playing_piano", "242": "playing_poker", "243": "playing_recorder", "244": "playing_saxophone", "245": "playing_squash_or_racquetball", "246": "playing_tennis", "247": "playing_trombone", "248": "playing_trumpet", "249": "playing_ukulele", "250": "playing_violin", "251": "playing_volleyball", "252": "playing_xylophone", "253": "pole_vault", "254": "presenting_weather_forecast", "255": "pull_ups", "256": "pumping_fist", "257": "pumping_gas", "258": "punching_bag", "259": "punching_person_(boxing)", "260": "push_up", "261": "pushing_car", "262": "pushing_cart", "263": "pushing_wheelchair", "264": "reading_book", "265": "reading_newspaper", "266": "recording_music", "267": "riding_a_bike", "268": "riding_camel", "269": "riding_elephant", "270": "riding_mechanical_bull", "271": "riding_mountain_bike", "272": "riding_mule", "273": "riding_or_walking_with_horse", "274": "riding_scooter", "275": "riding_unicycle", "276": "ripping_paper", "277": "robot_dancing", "278": "rock_climbing", "279": "rock_scissors_paper", "280": "roller_skating", "281": "running_on_treadmill", "282": "sailing", "283": "salsa_dancing", "284": "sanding_floor", "285": "scrambling_eggs", "286": "scuba_diving", "287": "setting_table", "288": "shaking_hands", "289": "shaking_head", "290": "sharpening_knives", "291": "sharpening_pencil", "292": "shaving_head", "293": "shaving_legs", "294": "shearing_sheep", "295": "shining_shoes", "296": "shooting_basketball", "297": "shooting_goal_(soccer)", "298": "shot_put", "299": "shoveling_snow", "300": "shredding_paper", "301": "shuffling_cards", "302": "side_kick", "303": "sign_language_interpreting", "304": "singing", "305": "situp", "306": "skateboarding", "307": "ski_jumping", "308": "skiing_(not_slalom_or_crosscountry)", "309": "skiing_crosscountry", "310": "skiing_slalom", "311": "skipping_rope", "312": "skydiving", "313": "slacklining", "314": "slapping", "315": "sled_dog_racing", "316": "smoking", "317": "smoking_hookah", "318": "snatch_weight_lifting", "319": "sneezing", "320": "sniffing", "321": "snorkeling", "322": "snowboarding", "323": "snowkiting", "324": "snowmobiling", "325": "somersaulting", "326": "spinning_poi", "327": "spray_painting", "328": "spraying", "329": "springboard_diving", "330": "squat", "331": "sticking_tongue_out", "332": "stomping_grapes", "333": "stretching_arm", "334": "stretching_leg", "335": "strumming_guitar", "336": "surfing_crowd", "337": "surfing_water", "338": "sweeping_floor", "339": "swimming_backstroke", "340": "swimming_breast_stroke", "341": "swimming_butterfly_stroke", "342": "swing_dancing", "343": "swinging_legs", "344": "swinging_on_something", "345": "sword_fighting", "346": "tai_chi", "347": "taking_a_shower", "348": "tango_dancing", "349": "tap_dancing", "350": "tapping_guitar", "351": "tapping_pen", "352": "tasting_beer", "353": "tasting_food", "354": "testifying", "355": "texting", "356": "throwing_axe", "357": "throwing_ball", "358": "throwing_discus", "359": "tickling", "360": "tobogganing", "361": "tossing_coin", "362": "tossing_salad", "363": "training_dog", "364": "trapezing", "365": "trimming_or_shaving_beard", "366": "trimming_trees", "367": "triple_jump", "368": "tying_bow_tie", "369": "tying_knot_(not_on_a_tie)", "370": "tying_tie", "371": "unboxing", "372": "unloading_truck", "373": "using_computer", "374": "using_remote_controller_(not_gaming)", "375": "using_segway", "376": "vault", "377": "waiting_in_line", "378": "walking_the_dog", "379": "washing_dishes", "380": "washing_feet", "381": "washing_hair", "382": "washing_hands", "383": "water_skiing", "384": "water_sliding", "385": "watering_plants", "386": "waxing_back", "387": "waxing_chest", "388": "waxing_eyebrows", "389": "waxing_legs", "390": "weaving_basket", "391": "welding", "392": "whistling", "393": "windsurfing", "394": "wrapping_present", "395": "wrestling", "396": "writing", "397": "yawning", "398": "yoga", "399": "zumba"} \ No newline at end of file +0 abseiling +1 air_drumming +2 answering_questions +3 applauding +4 applying_cream +5 archery +6 arm_wrestling +7 arranging_flowers +8 assembling_computer +9 auctioning +10 baby_waking_up +11 baking_cookies +12 balloon_blowing +13 bandaging +14 barbequing +15 bartending +16 beatboxing +17 bee_keeping +18 belly_dancing +19 bench_pressing +20 bending_back +21 bending_metal +22 biking_through_snow +23 blasting_sand +24 blowing_glass +25 blowing_leaves +26 blowing_nose +27 blowing_out_candles +28 bobsledding +29 bookbinding +30 bouncing_on_trampoline +31 bowling +32 braiding_hair +33 breading_or_breadcrumbing +34 breakdancing +35 brush_painting +36 brushing_hair +37 brushing_teeth +38 building_cabinet +39 building_shed +40 bungee_jumping +41 busking +42 canoeing_or_kayaking +43 capoeira +44 carrying_baby +45 cartwheeling +46 carving_pumpkin +47 catching_fish +48 catching_or_throwing_baseball +49 catching_or_throwing_frisbee +50 catching_or_throwing_softball +51 celebrating +52 changing_oil +53 changing_wheel +54 checking_tires +55 cheerleading +56 chopping_wood +57 clapping +58 clay_pottery_making +59 clean_and_jerk +60 cleaning_floor +61 cleaning_gutters +62 cleaning_pool +63 cleaning_shoes +64 cleaning_toilet +65 cleaning_windows +66 climbing_a_rope +67 climbing_ladder +68 climbing_tree +69 contact_juggling +70 cooking_chicken +71 cooking_egg +72 cooking_on_campfire +73 cooking_sausages +74 counting_money +75 country_line_dancing +76 cracking_neck +77 crawling_baby +78 crossing_river +79 crying +80 curling_hair +81 cutting_nails +82 cutting_pineapple +83 cutting_watermelon +84 dancing_ballet +85 dancing_charleston +86 dancing_gangnam_style +87 dancing_macarena +88 deadlifting +89 decorating_the_christmas_tree +90 digging +91 dining +92 disc_golfing +93 diving_cliff +94 dodgeball +95 doing_aerobics +96 doing_laundry +97 doing_nails +98 drawing +99 dribbling_basketball +100 drinking +101 drinking_beer +102 drinking_shots +103 driving_car +104 driving_tractor +105 drop_kicking +106 drumming_fingers +107 dunking_basketball +108 dying_hair +109 eating_burger +110 eating_cake +111 eating_carrots +112 eating_chips +113 eating_doughnuts +114 eating_hotdog +115 eating_ice_cream +116 eating_spaghetti +117 eating_watermelon +118 egg_hunting +119 exercising_arm +120 exercising_with_an_exercise_ball +121 extinguishing_fire +122 faceplanting +123 feeding_birds +124 feeding_fish +125 feeding_goats +126 filling_eyebrows +127 finger_snapping +128 fixing_hair +129 flipping_pancake +130 flying_kite +131 folding_clothes +132 folding_napkins +133 folding_paper +134 front_raises +135 frying_vegetables +136 garbage_collecting +137 gargling +138 getting_a_haircut +139 getting_a_tattoo +140 giving_or_receiving_award +141 golf_chipping +142 golf_driving +143 golf_putting +144 grinding_meat +145 grooming_dog +146 grooming_horse +147 gymnastics_tumbling +148 hammer_throw +149 headbanging +150 headbutting +151 high_jump +152 high_kick +153 hitting_baseball +154 hockey_stop +155 holding_snake +156 hopscotch +157 hoverboarding +158 hugging +159 hula_hooping +160 hurdling +161 hurling_(sport) +162 ice_climbing +163 ice_fishing +164 ice_skating +165 ironing +166 javelin_throw +167 jetskiing +168 jogging +169 juggling_balls +170 juggling_fire +171 juggling_soccer_ball +172 jumping_into_pool +173 jumpstyle_dancing +174 kicking_field_goal +175 kicking_soccer_ball +176 kissing +177 kitesurfing +178 knitting +179 krumping +180 laughing +181 laying_bricks +182 long_jump +183 lunge +184 making_a_cake +185 making_a_sandwich +186 making_bed +187 making_jewelry +188 making_pizza +189 making_snowman +190 making_sushi +191 making_tea +192 marching +193 massaging_back +194 massaging_feet +195 massaging_legs +196 massaging_person's_head +197 milking_cow +198 mopping_floor +199 motorcycling +200 moving_furniture +201 mowing_lawn +202 news_anchoring +203 opening_bottle +204 opening_present +205 paragliding +206 parasailing +207 parkour +208 passing_American_football_(in_game) +209 passing_American_football_(not_in_game) +210 peeling_apples +211 peeling_potatoes +212 petting_animal_(not_cat) +213 petting_cat +214 picking_fruit +215 planting_trees +216 plastering +217 playing_accordion +218 playing_badminton +219 playing_bagpipes +220 playing_basketball +221 playing_bass_guitar +222 playing_cards +223 playing_cello +224 playing_chess +225 playing_clarinet +226 playing_controller +227 playing_cricket +228 playing_cymbals +229 playing_didgeridoo +230 playing_drums +231 playing_flute +232 playing_guitar +233 playing_harmonica +234 playing_harp +235 playing_ice_hockey +236 playing_keyboard +237 playing_kickball +238 playing_monopoly +239 playing_organ +240 playing_paintball +241 playing_piano +242 playing_poker +243 playing_recorder +244 playing_saxophone +245 playing_squash_or_racquetball +246 playing_tennis +247 playing_trombone +248 playing_trumpet +249 playing_ukulele +250 playing_violin +251 playing_volleyball +252 playing_xylophone +253 pole_vault +254 presenting_weather_forecast +255 pull_ups +256 pumping_fist +257 pumping_gas +258 punching_bag +259 punching_person_(boxing) +260 push_up +261 pushing_car +262 pushing_cart +263 pushing_wheelchair +264 reading_book +265 reading_newspaper +266 recording_music +267 riding_a_bike +268 riding_camel +269 riding_elephant +270 riding_mechanical_bull +271 riding_mountain_bike +272 riding_mule +273 riding_or_walking_with_horse +274 riding_scooter +275 riding_unicycle +276 ripping_paper +277 robot_dancing +278 rock_climbing +279 rock_scissors_paper +280 roller_skating +281 running_on_treadmill +282 sailing +283 salsa_dancing +284 sanding_floor +285 scrambling_eggs +286 scuba_diving +287 setting_table +288 shaking_hands +289 shaking_head +290 sharpening_knives +291 sharpening_pencil +292 shaving_head +293 shaving_legs +294 shearing_sheep +295 shining_shoes +296 shooting_basketball +297 shooting_goal_(soccer) +298 shot_put +299 shoveling_snow +300 shredding_paper +301 shuffling_cards +302 side_kick +303 sign_language_interpreting +304 singing +305 situp +306 skateboarding +307 ski_jumping +308 skiing_(not_slalom_or_crosscountry) +309 skiing_crosscountry +310 skiing_slalom +311 skipping_rope +312 skydiving +313 slacklining +314 slapping +315 sled_dog_racing +316 smoking +317 smoking_hookah +318 snatch_weight_lifting +319 sneezing +320 sniffing +321 snorkeling +322 snowboarding +323 snowkiting +324 snowmobiling +325 somersaulting +326 spinning_poi +327 spray_painting +328 spraying +329 springboard_diving +330 squat +331 sticking_tongue_out +332 stomping_grapes +333 stretching_arm +334 stretching_leg +335 strumming_guitar +336 surfing_crowd +337 surfing_water +338 sweeping_floor +339 swimming_backstroke +340 swimming_breast_stroke +341 swimming_butterfly_stroke +342 swing_dancing +343 swinging_legs +344 swinging_on_something +345 sword_fighting +346 tai_chi +347 taking_a_shower +348 tango_dancing +349 tap_dancing +350 tapping_guitar +351 tapping_pen +352 tasting_beer +353 tasting_food +354 testifying +355 texting +356 throwing_axe +357 throwing_ball +358 throwing_discus +359 tickling +360 tobogganing +361 tossing_coin +362 tossing_salad +363 training_dog +364 trapezing +365 trimming_or_shaving_beard +366 trimming_trees +367 triple_jump +368 tying_bow_tie +369 tying_knot_(not_on_a_tie) +370 tying_tie +371 unboxing +372 unloading_truck +373 using_computer +374 using_remote_controller_(not_gaming) +375 using_segway +376 vault +377 waiting_in_line +378 walking_the_dog +379 washing_dishes +380 washing_feet +381 washing_hair +382 washing_hands +383 water_skiing +384 water_sliding +385 watering_plants +386 waxing_back +387 waxing_chest +388 waxing_eyebrows +389 waxing_legs +390 weaving_basket +391 welding +392 whistling +393 windsurfing +394 wrapping_present +395 wrestling +396 writing +397 yawning +398 yoga +399 zumba diff --git a/tools/paddlevideo_clas.py b/tools/paddlevideo_clas.py index fe2d53b52..bdef176c7 100644 --- a/tools/paddlevideo_clas.py +++ b/tools/paddlevideo_clas.py @@ -128,13 +128,13 @@ def load_label_name_dict(path): 'Warning: If want to use your own label_dict, please input legal path!\nOtherwise label_names will be empty!' ) else: - try: - import json - f = open(path,'r') - content = f.read() - result = json.loads(content) - except: - result = {} + for line in open(path, 'r'): + partition = line.split('\n')[0].partition(' ') + try: + result[int(partition[0])] = str(partition[-1]) + except: + result = {} + break return result def parse_args(mMain=True, add_help=True): From daf87258ac28ade4bfe872f123f740e623ba1646 Mon Sep 17 00:00:00 2001 From: xt <2761564455@qq.com> Date: Thu, 14 Jan 2021 11:48:55 +0800 Subject: [PATCH 5/7] update MANFEST.in --- MANIFEST.in | 1 + 1 file changed, 1 insertion(+) diff --git a/MANIFEST.in b/MANIFEST.in index cf4711257..f7df77b99 100644 --- a/MANIFEST.in +++ b/MANIFEST.in @@ -3,5 +3,6 @@ include README.md include tools/__init__.py include tools/utils.py include tools/paddlevideo_clas.py +include tools/Kinetics-400_label_list.txt recursive-include paddlevideo/ *.py *.txt \ No newline at end of file From b0732456fc84986f6f3b9f1256293154b66597b1 Mon Sep 17 00:00:00 2001 From: xt <2761564455@qq.com> Date: Thu, 14 Jan 2021 18:41:14 +0800 Subject: [PATCH 6/7] mv k400_label_list.txt to data/k400 --- MANIFEST.in | 2 +- setup.py | 4 +- tools/Kinetics-400_label_list.txt | 400 ------------------------------ tools/paddlevideo_clas.py | 2 +- 4 files changed, 4 insertions(+), 404 deletions(-) delete mode 100644 tools/Kinetics-400_label_list.txt diff --git a/MANIFEST.in b/MANIFEST.in index f7df77b99..4a5741053 100644 --- a/MANIFEST.in +++ b/MANIFEST.in @@ -3,6 +3,6 @@ include README.md include tools/__init__.py include tools/utils.py include tools/paddlevideo_clas.py -include tools/Kinetics-400_label_list.txt +include data/k400/Kinetics-400_label_list.txt recursive-include paddlevideo/ *.py *.txt \ No newline at end of file diff --git a/setup.py b/setup.py index de05c4415..2b1936039 100644 --- a/setup.py +++ b/setup.py @@ -24,12 +24,12 @@ def readme(): setup( - name='ppvideo', #name of .whl file + name='paddlevideo', #name of .whl file packages=['ppvideo'], #install package name package_dir={'ppvideo': ''}, include_package_data=True, #Accept all data files and directories matched by MANIFEST.in install_requires=requirements, - version='2.0.0', + version='0.0.0', license='Apache License 2.0', description='Awesome Video toolkits based on PaddlePaddle ', long_description=readme(), diff --git a/tools/Kinetics-400_label_list.txt b/tools/Kinetics-400_label_list.txt deleted file mode 100644 index 8488908b1..000000000 --- a/tools/Kinetics-400_label_list.txt +++ /dev/null @@ -1,400 +0,0 @@ -0 abseiling -1 air_drumming -2 answering_questions -3 applauding -4 applying_cream -5 archery -6 arm_wrestling -7 arranging_flowers -8 assembling_computer -9 auctioning -10 baby_waking_up -11 baking_cookies -12 balloon_blowing -13 bandaging -14 barbequing -15 bartending -16 beatboxing -17 bee_keeping -18 belly_dancing -19 bench_pressing -20 bending_back -21 bending_metal -22 biking_through_snow -23 blasting_sand -24 blowing_glass -25 blowing_leaves -26 blowing_nose -27 blowing_out_candles -28 bobsledding -29 bookbinding -30 bouncing_on_trampoline -31 bowling -32 braiding_hair -33 breading_or_breadcrumbing -34 breakdancing -35 brush_painting -36 brushing_hair -37 brushing_teeth -38 building_cabinet -39 building_shed -40 bungee_jumping -41 busking -42 canoeing_or_kayaking -43 capoeira -44 carrying_baby -45 cartwheeling -46 carving_pumpkin -47 catching_fish -48 catching_or_throwing_baseball -49 catching_or_throwing_frisbee -50 catching_or_throwing_softball -51 celebrating -52 changing_oil -53 changing_wheel -54 checking_tires -55 cheerleading -56 chopping_wood -57 clapping -58 clay_pottery_making -59 clean_and_jerk -60 cleaning_floor -61 cleaning_gutters -62 cleaning_pool -63 cleaning_shoes -64 cleaning_toilet -65 cleaning_windows -66 climbing_a_rope -67 climbing_ladder -68 climbing_tree -69 contact_juggling -70 cooking_chicken -71 cooking_egg -72 cooking_on_campfire -73 cooking_sausages -74 counting_money -75 country_line_dancing -76 cracking_neck -77 crawling_baby -78 crossing_river -79 crying -80 curling_hair -81 cutting_nails -82 cutting_pineapple -83 cutting_watermelon -84 dancing_ballet -85 dancing_charleston -86 dancing_gangnam_style -87 dancing_macarena -88 deadlifting -89 decorating_the_christmas_tree -90 digging -91 dining -92 disc_golfing -93 diving_cliff -94 dodgeball -95 doing_aerobics -96 doing_laundry -97 doing_nails -98 drawing -99 dribbling_basketball -100 drinking -101 drinking_beer -102 drinking_shots -103 driving_car -104 driving_tractor -105 drop_kicking -106 drumming_fingers -107 dunking_basketball -108 dying_hair -109 eating_burger -110 eating_cake -111 eating_carrots -112 eating_chips -113 eating_doughnuts -114 eating_hotdog -115 eating_ice_cream -116 eating_spaghetti -117 eating_watermelon -118 egg_hunting -119 exercising_arm -120 exercising_with_an_exercise_ball -121 extinguishing_fire -122 faceplanting -123 feeding_birds -124 feeding_fish -125 feeding_goats -126 filling_eyebrows -127 finger_snapping -128 fixing_hair -129 flipping_pancake -130 flying_kite -131 folding_clothes -132 folding_napkins -133 folding_paper -134 front_raises -135 frying_vegetables -136 garbage_collecting -137 gargling -138 getting_a_haircut -139 getting_a_tattoo -140 giving_or_receiving_award -141 golf_chipping -142 golf_driving -143 golf_putting -144 grinding_meat -145 grooming_dog -146 grooming_horse -147 gymnastics_tumbling -148 hammer_throw -149 headbanging -150 headbutting -151 high_jump -152 high_kick -153 hitting_baseball -154 hockey_stop -155 holding_snake -156 hopscotch -157 hoverboarding -158 hugging -159 hula_hooping -160 hurdling -161 hurling_(sport) -162 ice_climbing -163 ice_fishing -164 ice_skating -165 ironing -166 javelin_throw -167 jetskiing -168 jogging -169 juggling_balls -170 juggling_fire -171 juggling_soccer_ball -172 jumping_into_pool -173 jumpstyle_dancing -174 kicking_field_goal -175 kicking_soccer_ball -176 kissing -177 kitesurfing -178 knitting -179 krumping -180 laughing -181 laying_bricks -182 long_jump -183 lunge -184 making_a_cake -185 making_a_sandwich -186 making_bed -187 making_jewelry -188 making_pizza -189 making_snowman -190 making_sushi -191 making_tea -192 marching -193 massaging_back -194 massaging_feet -195 massaging_legs -196 massaging_person's_head -197 milking_cow -198 mopping_floor -199 motorcycling -200 moving_furniture -201 mowing_lawn -202 news_anchoring -203 opening_bottle -204 opening_present -205 paragliding -206 parasailing -207 parkour -208 passing_American_football_(in_game) -209 passing_American_football_(not_in_game) -210 peeling_apples -211 peeling_potatoes -212 petting_animal_(not_cat) -213 petting_cat -214 picking_fruit -215 planting_trees -216 plastering -217 playing_accordion -218 playing_badminton -219 playing_bagpipes -220 playing_basketball -221 playing_bass_guitar -222 playing_cards -223 playing_cello -224 playing_chess -225 playing_clarinet -226 playing_controller -227 playing_cricket -228 playing_cymbals -229 playing_didgeridoo -230 playing_drums -231 playing_flute -232 playing_guitar -233 playing_harmonica -234 playing_harp -235 playing_ice_hockey -236 playing_keyboard -237 playing_kickball -238 playing_monopoly -239 playing_organ -240 playing_paintball -241 playing_piano -242 playing_poker -243 playing_recorder -244 playing_saxophone -245 playing_squash_or_racquetball -246 playing_tennis -247 playing_trombone -248 playing_trumpet -249 playing_ukulele -250 playing_violin -251 playing_volleyball -252 playing_xylophone -253 pole_vault -254 presenting_weather_forecast -255 pull_ups -256 pumping_fist -257 pumping_gas -258 punching_bag -259 punching_person_(boxing) -260 push_up -261 pushing_car -262 pushing_cart -263 pushing_wheelchair -264 reading_book -265 reading_newspaper -266 recording_music -267 riding_a_bike -268 riding_camel -269 riding_elephant -270 riding_mechanical_bull -271 riding_mountain_bike -272 riding_mule -273 riding_or_walking_with_horse -274 riding_scooter -275 riding_unicycle -276 ripping_paper -277 robot_dancing -278 rock_climbing -279 rock_scissors_paper -280 roller_skating -281 running_on_treadmill -282 sailing -283 salsa_dancing -284 sanding_floor -285 scrambling_eggs -286 scuba_diving -287 setting_table -288 shaking_hands -289 shaking_head -290 sharpening_knives -291 sharpening_pencil -292 shaving_head -293 shaving_legs -294 shearing_sheep -295 shining_shoes -296 shooting_basketball -297 shooting_goal_(soccer) -298 shot_put -299 shoveling_snow -300 shredding_paper -301 shuffling_cards -302 side_kick -303 sign_language_interpreting -304 singing -305 situp -306 skateboarding -307 ski_jumping -308 skiing_(not_slalom_or_crosscountry) -309 skiing_crosscountry -310 skiing_slalom -311 skipping_rope -312 skydiving -313 slacklining -314 slapping -315 sled_dog_racing -316 smoking -317 smoking_hookah -318 snatch_weight_lifting -319 sneezing -320 sniffing -321 snorkeling -322 snowboarding -323 snowkiting -324 snowmobiling -325 somersaulting -326 spinning_poi -327 spray_painting -328 spraying -329 springboard_diving -330 squat -331 sticking_tongue_out -332 stomping_grapes -333 stretching_arm -334 stretching_leg -335 strumming_guitar -336 surfing_crowd -337 surfing_water -338 sweeping_floor -339 swimming_backstroke -340 swimming_breast_stroke -341 swimming_butterfly_stroke -342 swing_dancing -343 swinging_legs -344 swinging_on_something -345 sword_fighting -346 tai_chi -347 taking_a_shower -348 tango_dancing -349 tap_dancing -350 tapping_guitar -351 tapping_pen -352 tasting_beer -353 tasting_food -354 testifying -355 texting -356 throwing_axe -357 throwing_ball -358 throwing_discus -359 tickling -360 tobogganing -361 tossing_coin -362 tossing_salad -363 training_dog -364 trapezing -365 trimming_or_shaving_beard -366 trimming_trees -367 triple_jump -368 tying_bow_tie -369 tying_knot_(not_on_a_tie) -370 tying_tie -371 unboxing -372 unloading_truck -373 using_computer -374 using_remote_controller_(not_gaming) -375 using_segway -376 vault -377 waiting_in_line -378 walking_the_dog -379 washing_dishes -380 washing_feet -381 washing_hair -382 washing_hands -383 water_skiing -384 water_sliding -385 watering_plants -386 waxing_back -387 waxing_chest -388 waxing_eyebrows -389 waxing_legs -390 weaving_basket -391 welding -392 whistling -393 windsurfing -394 wrapping_present -395 wrestling -396 writing -397 yawning -398 yoga -399 zumba diff --git a/tools/paddlevideo_clas.py b/tools/paddlevideo_clas.py index bdef176c7..3f163d5c4 100644 --- a/tools/paddlevideo_clas.py +++ b/tools/paddlevideo_clas.py @@ -240,7 +240,7 @@ def __init__(self, **kwargs): process_params.params_file = os.path.join( download_path, 'inference.pdiparams') process_params.label_name_path = os.path.join( - __dir__, 'Kinetics-400_label_list.txt') + __dir__, '../data/k400/Kinetics-400_label_list.txt') else: raise Exception( 'If you want to use your own model, Please input model_file as model path!' From 41790030f2acd776540f32da1fd2d69dfaa3016e Mon Sep 17 00:00:00 2001 From: xt <2761564455@qq.com> Date: Thu, 14 Jan 2021 18:59:24 +0800 Subject: [PATCH 7/7] update k400-label --- data/k400/Kinetics-400_label_list.txt | 400 ++++++++++++++++++++++++++ 1 file changed, 400 insertions(+) create mode 100644 data/k400/Kinetics-400_label_list.txt diff --git a/data/k400/Kinetics-400_label_list.txt b/data/k400/Kinetics-400_label_list.txt new file mode 100644 index 000000000..8488908b1 --- /dev/null +++ b/data/k400/Kinetics-400_label_list.txt @@ -0,0 +1,400 @@ +0 abseiling +1 air_drumming +2 answering_questions +3 applauding +4 applying_cream +5 archery +6 arm_wrestling +7 arranging_flowers +8 assembling_computer +9 auctioning +10 baby_waking_up +11 baking_cookies +12 balloon_blowing +13 bandaging +14 barbequing +15 bartending +16 beatboxing +17 bee_keeping +18 belly_dancing +19 bench_pressing +20 bending_back +21 bending_metal +22 biking_through_snow +23 blasting_sand +24 blowing_glass +25 blowing_leaves +26 blowing_nose +27 blowing_out_candles +28 bobsledding +29 bookbinding +30 bouncing_on_trampoline +31 bowling +32 braiding_hair +33 breading_or_breadcrumbing +34 breakdancing +35 brush_painting +36 brushing_hair +37 brushing_teeth +38 building_cabinet +39 building_shed +40 bungee_jumping +41 busking +42 canoeing_or_kayaking +43 capoeira +44 carrying_baby +45 cartwheeling +46 carving_pumpkin +47 catching_fish +48 catching_or_throwing_baseball +49 catching_or_throwing_frisbee +50 catching_or_throwing_softball +51 celebrating +52 changing_oil +53 changing_wheel +54 checking_tires +55 cheerleading +56 chopping_wood +57 clapping +58 clay_pottery_making +59 clean_and_jerk +60 cleaning_floor +61 cleaning_gutters +62 cleaning_pool +63 cleaning_shoes +64 cleaning_toilet +65 cleaning_windows +66 climbing_a_rope +67 climbing_ladder +68 climbing_tree +69 contact_juggling +70 cooking_chicken +71 cooking_egg +72 cooking_on_campfire +73 cooking_sausages +74 counting_money +75 country_line_dancing +76 cracking_neck +77 crawling_baby +78 crossing_river +79 crying +80 curling_hair +81 cutting_nails +82 cutting_pineapple +83 cutting_watermelon +84 dancing_ballet +85 dancing_charleston +86 dancing_gangnam_style +87 dancing_macarena +88 deadlifting +89 decorating_the_christmas_tree +90 digging +91 dining +92 disc_golfing +93 diving_cliff +94 dodgeball +95 doing_aerobics +96 doing_laundry +97 doing_nails +98 drawing +99 dribbling_basketball +100 drinking +101 drinking_beer +102 drinking_shots +103 driving_car +104 driving_tractor +105 drop_kicking +106 drumming_fingers +107 dunking_basketball +108 dying_hair +109 eating_burger +110 eating_cake +111 eating_carrots +112 eating_chips +113 eating_doughnuts +114 eating_hotdog +115 eating_ice_cream +116 eating_spaghetti +117 eating_watermelon +118 egg_hunting +119 exercising_arm +120 exercising_with_an_exercise_ball +121 extinguishing_fire +122 faceplanting +123 feeding_birds +124 feeding_fish +125 feeding_goats +126 filling_eyebrows +127 finger_snapping +128 fixing_hair +129 flipping_pancake +130 flying_kite +131 folding_clothes +132 folding_napkins +133 folding_paper +134 front_raises +135 frying_vegetables +136 garbage_collecting +137 gargling +138 getting_a_haircut +139 getting_a_tattoo +140 giving_or_receiving_award +141 golf_chipping +142 golf_driving +143 golf_putting +144 grinding_meat +145 grooming_dog +146 grooming_horse +147 gymnastics_tumbling +148 hammer_throw +149 headbanging +150 headbutting +151 high_jump +152 high_kick +153 hitting_baseball +154 hockey_stop +155 holding_snake +156 hopscotch +157 hoverboarding +158 hugging +159 hula_hooping +160 hurdling +161 hurling_(sport) +162 ice_climbing +163 ice_fishing +164 ice_skating +165 ironing +166 javelin_throw +167 jetskiing +168 jogging +169 juggling_balls +170 juggling_fire +171 juggling_soccer_ball +172 jumping_into_pool +173 jumpstyle_dancing +174 kicking_field_goal +175 kicking_soccer_ball +176 kissing +177 kitesurfing +178 knitting +179 krumping +180 laughing +181 laying_bricks +182 long_jump +183 lunge +184 making_a_cake +185 making_a_sandwich +186 making_bed +187 making_jewelry +188 making_pizza +189 making_snowman +190 making_sushi +191 making_tea +192 marching +193 massaging_back +194 massaging_feet +195 massaging_legs +196 massaging_person's_head +197 milking_cow +198 mopping_floor +199 motorcycling +200 moving_furniture +201 mowing_lawn +202 news_anchoring +203 opening_bottle +204 opening_present +205 paragliding +206 parasailing +207 parkour +208 passing_American_football_(in_game) +209 passing_American_football_(not_in_game) +210 peeling_apples +211 peeling_potatoes +212 petting_animal_(not_cat) +213 petting_cat +214 picking_fruit +215 planting_trees +216 plastering +217 playing_accordion +218 playing_badminton +219 playing_bagpipes +220 playing_basketball +221 playing_bass_guitar +222 playing_cards +223 playing_cello +224 playing_chess +225 playing_clarinet +226 playing_controller +227 playing_cricket +228 playing_cymbals +229 playing_didgeridoo +230 playing_drums +231 playing_flute +232 playing_guitar +233 playing_harmonica +234 playing_harp +235 playing_ice_hockey +236 playing_keyboard +237 playing_kickball +238 playing_monopoly +239 playing_organ +240 playing_paintball +241 playing_piano +242 playing_poker +243 playing_recorder +244 playing_saxophone +245 playing_squash_or_racquetball +246 playing_tennis +247 playing_trombone +248 playing_trumpet +249 playing_ukulele +250 playing_violin +251 playing_volleyball +252 playing_xylophone +253 pole_vault +254 presenting_weather_forecast +255 pull_ups +256 pumping_fist +257 pumping_gas +258 punching_bag +259 punching_person_(boxing) +260 push_up +261 pushing_car +262 pushing_cart +263 pushing_wheelchair +264 reading_book +265 reading_newspaper +266 recording_music +267 riding_a_bike +268 riding_camel +269 riding_elephant +270 riding_mechanical_bull +271 riding_mountain_bike +272 riding_mule +273 riding_or_walking_with_horse +274 riding_scooter +275 riding_unicycle +276 ripping_paper +277 robot_dancing +278 rock_climbing +279 rock_scissors_paper +280 roller_skating +281 running_on_treadmill +282 sailing +283 salsa_dancing +284 sanding_floor +285 scrambling_eggs +286 scuba_diving +287 setting_table +288 shaking_hands +289 shaking_head +290 sharpening_knives +291 sharpening_pencil +292 shaving_head +293 shaving_legs +294 shearing_sheep +295 shining_shoes +296 shooting_basketball +297 shooting_goal_(soccer) +298 shot_put +299 shoveling_snow +300 shredding_paper +301 shuffling_cards +302 side_kick +303 sign_language_interpreting +304 singing +305 situp +306 skateboarding +307 ski_jumping +308 skiing_(not_slalom_or_crosscountry) +309 skiing_crosscountry +310 skiing_slalom +311 skipping_rope +312 skydiving +313 slacklining +314 slapping +315 sled_dog_racing +316 smoking +317 smoking_hookah +318 snatch_weight_lifting +319 sneezing +320 sniffing +321 snorkeling +322 snowboarding +323 snowkiting +324 snowmobiling +325 somersaulting +326 spinning_poi +327 spray_painting +328 spraying +329 springboard_diving +330 squat +331 sticking_tongue_out +332 stomping_grapes +333 stretching_arm +334 stretching_leg +335 strumming_guitar +336 surfing_crowd +337 surfing_water +338 sweeping_floor +339 swimming_backstroke +340 swimming_breast_stroke +341 swimming_butterfly_stroke +342 swing_dancing +343 swinging_legs +344 swinging_on_something +345 sword_fighting +346 tai_chi +347 taking_a_shower +348 tango_dancing +349 tap_dancing +350 tapping_guitar +351 tapping_pen +352 tasting_beer +353 tasting_food +354 testifying +355 texting +356 throwing_axe +357 throwing_ball +358 throwing_discus +359 tickling +360 tobogganing +361 tossing_coin +362 tossing_salad +363 training_dog +364 trapezing +365 trimming_or_shaving_beard +366 trimming_trees +367 triple_jump +368 tying_bow_tie +369 tying_knot_(not_on_a_tie) +370 tying_tie +371 unboxing +372 unloading_truck +373 using_computer +374 using_remote_controller_(not_gaming) +375 using_segway +376 vault +377 waiting_in_line +378 walking_the_dog +379 washing_dishes +380 washing_feet +381 washing_hair +382 washing_hands +383 water_skiing +384 water_sliding +385 watering_plants +386 waxing_back +387 waxing_chest +388 waxing_eyebrows +389 waxing_legs +390 weaving_basket +391 welding +392 whistling +393 windsurfing +394 wrapping_present +395 wrestling +396 writing +397 yawning +398 yoga +399 zumba