Step 1. Action recognition (modified. original code source: https://github.com/jfzhang95/pytorch-video-recognition)
-
Accuracy
Train accuracy over 99%
Top validation accuracy over 97%
Step 2. Localization
- Using YOLO v3 (modified. original code source: https://github.com/ayooshkathuria/pytorch-yolo-v3)
Download APS dataset
Unzip dataset in root directory.
python split_video.py aps_original aps_cut
will make a new folder named 'aps_cut' and put all split videos there.
python train.py
will make a new folder named 'data' and start training.
Training Results are like below.
validation accuracy
validation loss
Go to yolov3 directory.
Download pre-trained weight here
python video.py --video video_input/aps.avi video_output/