├── .gitignore ├── README.md ├── config ├── __init__.py ├── ava_categories_ratio.json ├── dataset_config.py └── yowof_config.py ├── dataset ├── __init__.py ├── ava.py ├── ava_helper.py ├── ava_pose.py ├── demo │ └── ucf24_demo │ │ ├── v_Basketball_g01_c02.mp4 │ │ ├── v_Basketball_g07_c04.mp4 │ │ ├── v_Biking_g01_c01.mp4 │ │ ├── v_CliffDiving_g03_c01.mp4 │ │ ├── v_Fencing_g01_c06.mp4 │ │ ├── v_HorseRiding_g01_c03.mp4 │ │ ├── v_IceDancing_g02_c05.mp4 │ │ ├── v_SalsaSpin_g03_c01.mp4 │ │ └── v_SkateBoarding_g02_c01.mp4 ├── transforms.py └── ucf_jhmdb.py ├── demo.py ├── eval.py ├── evaluator ├── __init__.py ├── ava_eval_helper.py ├── ava_evaluation │ ├── README.md │ ├── __init__.py │ ├── label_map_util.py │ ├── metrics.py │ ├── np_box_list.py │ ├── np_box_list_ops.py │ ├── np_box_mask_list.py │ ├── np_box_mask_list_ops.py │ ├── np_box_ops.py │ ├── np_mask_ops.py │ ├── object_detection_evaluation.py │ ├── per_image_evaluation.py │ └── standard_fields.py ├── ava_evaluator.py ├── cal_frame_mAP.py ├── cal_video_mAP.py ├── groundtruths_ucf_jhmdb │ ├── groundtruths_jhmdb.zip │ └── groundtruths_ucf.zip ├── ucf_jhmdb_evaluator.py └── utils.py ├── img_files ├── online_inference.png ├── v_Basketball_g01_c02.gif ├── v_Biking_g01_c01.gif ├── v_HorseRiding_g01_c03.gif ├── v_IceDancing_g02_c05.gif ├── v_SalsaSpin_g03_c01.gif ├── v_SkateBoarding_g02_c01.gif └── yowof.png ├── models ├── __init__.py ├── backbone │ ├── __init__.py │ ├── dla.py │ ├── resnet.py │ └── vgg.py ├── basic │ ├── conv.py │ └── convlstm.py ├── detector │ ├── __init__.py │ └── yowof │ │ ├── loss.py │ │ ├── matcher.py │ │ └── yowof.py └── head │ ├── __init__.py │ └── decoupled_head.py ├── requirements.txt ├── test.py ├── test_video_ava.py ├── train.py ├── train_ava.sh ├── train_ddp.sh ├── train_ucf.sh └── utils ├── __init__.py ├── box_ops.py ├── com_flops_params.py ├── distributed_utils.py ├── misc.py ├── solver ├── __init__.py ├── optimizer.py └── warmup_schedule.py ├── vis_tools.py └── weight_init.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yjh0410/YOWOF/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yjh0410/YOWOF/HEAD/README.md -------------------------------------------------------------------------------- /config/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yjh0410/YOWOF/HEAD/config/__init__.py -------------------------------------------------------------------------------- /config/ava_categories_ratio.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yjh0410/YOWOF/HEAD/config/ava_categories_ratio.json -------------------------------------------------------------------------------- /config/dataset_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yjh0410/YOWOF/HEAD/config/dataset_config.py -------------------------------------------------------------------------------- /config/yowof_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yjh0410/YOWOF/HEAD/config/yowof_config.py -------------------------------------------------------------------------------- /dataset/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dataset/ava.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yjh0410/YOWOF/HEAD/dataset/ava.py -------------------------------------------------------------------------------- /dataset/ava_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yjh0410/YOWOF/HEAD/dataset/ava_helper.py -------------------------------------------------------------------------------- /dataset/ava_pose.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yjh0410/YOWOF/HEAD/dataset/ava_pose.py -------------------------------------------------------------------------------- /dataset/demo/ucf24_demo/v_Basketball_g01_c02.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yjh0410/YOWOF/HEAD/dataset/demo/ucf24_demo/v_Basketball_g01_c02.mp4 -------------------------------------------------------------------------------- /dataset/demo/ucf24_demo/v_Basketball_g07_c04.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yjh0410/YOWOF/HEAD/dataset/demo/ucf24_demo/v_Basketball_g07_c04.mp4 -------------------------------------------------------------------------------- /dataset/demo/ucf24_demo/v_Biking_g01_c01.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yjh0410/YOWOF/HEAD/dataset/demo/ucf24_demo/v_Biking_g01_c01.mp4 -------------------------------------------------------------------------------- /dataset/demo/ucf24_demo/v_CliffDiving_g03_c01.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yjh0410/YOWOF/HEAD/dataset/demo/ucf24_demo/v_CliffDiving_g03_c01.mp4 -------------------------------------------------------------------------------- /dataset/demo/ucf24_demo/v_Fencing_g01_c06.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yjh0410/YOWOF/HEAD/dataset/demo/ucf24_demo/v_Fencing_g01_c06.mp4 -------------------------------------------------------------------------------- /dataset/demo/ucf24_demo/v_HorseRiding_g01_c03.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yjh0410/YOWOF/HEAD/dataset/demo/ucf24_demo/v_HorseRiding_g01_c03.mp4 -------------------------------------------------------------------------------- /dataset/demo/ucf24_demo/v_IceDancing_g02_c05.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yjh0410/YOWOF/HEAD/dataset/demo/ucf24_demo/v_IceDancing_g02_c05.mp4 -------------------------------------------------------------------------------- /dataset/demo/ucf24_demo/v_SalsaSpin_g03_c01.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yjh0410/YOWOF/HEAD/dataset/demo/ucf24_demo/v_SalsaSpin_g03_c01.mp4 -------------------------------------------------------------------------------- /dataset/demo/ucf24_demo/v_SkateBoarding_g02_c01.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yjh0410/YOWOF/HEAD/dataset/demo/ucf24_demo/v_SkateBoarding_g02_c01.mp4 -------------------------------------------------------------------------------- /dataset/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yjh0410/YOWOF/HEAD/dataset/transforms.py -------------------------------------------------------------------------------- /dataset/ucf_jhmdb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yjh0410/YOWOF/HEAD/dataset/ucf_jhmdb.py -------------------------------------------------------------------------------- /demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yjh0410/YOWOF/HEAD/demo.py -------------------------------------------------------------------------------- /eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yjh0410/YOWOF/HEAD/eval.py -------------------------------------------------------------------------------- /evaluator/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /evaluator/ava_eval_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yjh0410/YOWOF/HEAD/evaluator/ava_eval_helper.py -------------------------------------------------------------------------------- /evaluator/ava_evaluation/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yjh0410/YOWOF/HEAD/evaluator/ava_evaluation/README.md -------------------------------------------------------------------------------- /evaluator/ava_evaluation/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /evaluator/ava_evaluation/label_map_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yjh0410/YOWOF/HEAD/evaluator/ava_evaluation/label_map_util.py -------------------------------------------------------------------------------- /evaluator/ava_evaluation/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yjh0410/YOWOF/HEAD/evaluator/ava_evaluation/metrics.py -------------------------------------------------------------------------------- /evaluator/ava_evaluation/np_box_list.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yjh0410/YOWOF/HEAD/evaluator/ava_evaluation/np_box_list.py -------------------------------------------------------------------------------- /evaluator/ava_evaluation/np_box_list_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yjh0410/YOWOF/HEAD/evaluator/ava_evaluation/np_box_list_ops.py -------------------------------------------------------------------------------- /evaluator/ava_evaluation/np_box_mask_list.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yjh0410/YOWOF/HEAD/evaluator/ava_evaluation/np_box_mask_list.py -------------------------------------------------------------------------------- /evaluator/ava_evaluation/np_box_mask_list_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yjh0410/YOWOF/HEAD/evaluator/ava_evaluation/np_box_mask_list_ops.py -------------------------------------------------------------------------------- /evaluator/ava_evaluation/np_box_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yjh0410/YOWOF/HEAD/evaluator/ava_evaluation/np_box_ops.py -------------------------------------------------------------------------------- /evaluator/ava_evaluation/np_mask_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yjh0410/YOWOF/HEAD/evaluator/ava_evaluation/np_mask_ops.py -------------------------------------------------------------------------------- /evaluator/ava_evaluation/object_detection_evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yjh0410/YOWOF/HEAD/evaluator/ava_evaluation/object_detection_evaluation.py -------------------------------------------------------------------------------- /evaluator/ava_evaluation/per_image_evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yjh0410/YOWOF/HEAD/evaluator/ava_evaluation/per_image_evaluation.py -------------------------------------------------------------------------------- /evaluator/ava_evaluation/standard_fields.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yjh0410/YOWOF/HEAD/evaluator/ava_evaluation/standard_fields.py -------------------------------------------------------------------------------- /evaluator/ava_evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yjh0410/YOWOF/HEAD/evaluator/ava_evaluator.py -------------------------------------------------------------------------------- /evaluator/cal_frame_mAP.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yjh0410/YOWOF/HEAD/evaluator/cal_frame_mAP.py -------------------------------------------------------------------------------- /evaluator/cal_video_mAP.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yjh0410/YOWOF/HEAD/evaluator/cal_video_mAP.py -------------------------------------------------------------------------------- /evaluator/groundtruths_ucf_jhmdb/groundtruths_jhmdb.zip: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yjh0410/YOWOF/HEAD/evaluator/groundtruths_ucf_jhmdb/groundtruths_jhmdb.zip -------------------------------------------------------------------------------- /evaluator/groundtruths_ucf_jhmdb/groundtruths_ucf.zip: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yjh0410/YOWOF/HEAD/evaluator/groundtruths_ucf_jhmdb/groundtruths_ucf.zip -------------------------------------------------------------------------------- /evaluator/ucf_jhmdb_evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yjh0410/YOWOF/HEAD/evaluator/ucf_jhmdb_evaluator.py -------------------------------------------------------------------------------- /evaluator/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yjh0410/YOWOF/HEAD/evaluator/utils.py -------------------------------------------------------------------------------- /img_files/online_inference.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yjh0410/YOWOF/HEAD/img_files/online_inference.png -------------------------------------------------------------------------------- /img_files/v_Basketball_g01_c02.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yjh0410/YOWOF/HEAD/img_files/v_Basketball_g01_c02.gif -------------------------------------------------------------------------------- /img_files/v_Biking_g01_c01.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yjh0410/YOWOF/HEAD/img_files/v_Biking_g01_c01.gif -------------------------------------------------------------------------------- /img_files/v_HorseRiding_g01_c03.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yjh0410/YOWOF/HEAD/img_files/v_HorseRiding_g01_c03.gif -------------------------------------------------------------------------------- /img_files/v_IceDancing_g02_c05.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yjh0410/YOWOF/HEAD/img_files/v_IceDancing_g02_c05.gif -------------------------------------------------------------------------------- /img_files/v_SalsaSpin_g03_c01.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yjh0410/YOWOF/HEAD/img_files/v_SalsaSpin_g03_c01.gif -------------------------------------------------------------------------------- /img_files/v_SkateBoarding_g02_c01.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yjh0410/YOWOF/HEAD/img_files/v_SkateBoarding_g02_c01.gif -------------------------------------------------------------------------------- /img_files/yowof.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yjh0410/YOWOF/HEAD/img_files/yowof.png -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/backbone/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yjh0410/YOWOF/HEAD/models/backbone/__init__.py -------------------------------------------------------------------------------- /models/backbone/dla.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yjh0410/YOWOF/HEAD/models/backbone/dla.py -------------------------------------------------------------------------------- /models/backbone/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yjh0410/YOWOF/HEAD/models/backbone/resnet.py -------------------------------------------------------------------------------- /models/backbone/vgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yjh0410/YOWOF/HEAD/models/backbone/vgg.py -------------------------------------------------------------------------------- /models/basic/conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yjh0410/YOWOF/HEAD/models/basic/conv.py -------------------------------------------------------------------------------- /models/basic/convlstm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yjh0410/YOWOF/HEAD/models/basic/convlstm.py -------------------------------------------------------------------------------- /models/detector/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yjh0410/YOWOF/HEAD/models/detector/__init__.py -------------------------------------------------------------------------------- /models/detector/yowof/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yjh0410/YOWOF/HEAD/models/detector/yowof/loss.py -------------------------------------------------------------------------------- /models/detector/yowof/matcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yjh0410/YOWOF/HEAD/models/detector/yowof/matcher.py -------------------------------------------------------------------------------- /models/detector/yowof/yowof.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yjh0410/YOWOF/HEAD/models/detector/yowof/yowof.py -------------------------------------------------------------------------------- /models/head/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/head/decoupled_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yjh0410/YOWOF/HEAD/models/head/decoupled_head.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yjh0410/YOWOF/HEAD/requirements.txt -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yjh0410/YOWOF/HEAD/test.py -------------------------------------------------------------------------------- /test_video_ava.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yjh0410/YOWOF/HEAD/test_video_ava.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yjh0410/YOWOF/HEAD/train.py -------------------------------------------------------------------------------- /train_ava.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yjh0410/YOWOF/HEAD/train_ava.sh -------------------------------------------------------------------------------- /train_ddp.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yjh0410/YOWOF/HEAD/train_ddp.sh -------------------------------------------------------------------------------- /train_ucf.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yjh0410/YOWOF/HEAD/train_ucf.sh -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/box_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yjh0410/YOWOF/HEAD/utils/box_ops.py -------------------------------------------------------------------------------- /utils/com_flops_params.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yjh0410/YOWOF/HEAD/utils/com_flops_params.py -------------------------------------------------------------------------------- /utils/distributed_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yjh0410/YOWOF/HEAD/utils/distributed_utils.py -------------------------------------------------------------------------------- /utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yjh0410/YOWOF/HEAD/utils/misc.py -------------------------------------------------------------------------------- /utils/solver/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/solver/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yjh0410/YOWOF/HEAD/utils/solver/optimizer.py -------------------------------------------------------------------------------- /utils/solver/warmup_schedule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yjh0410/YOWOF/HEAD/utils/solver/warmup_schedule.py -------------------------------------------------------------------------------- /utils/vis_tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yjh0410/YOWOF/HEAD/utils/vis_tools.py -------------------------------------------------------------------------------- /utils/weight_init.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yjh0410/YOWOF/HEAD/utils/weight_init.py --------------------------------------------------------------------------------