Video feature extracor for your own datasets.
- Ubuntu 16.04
- CUDA 9.0
- cuDNN 7.5.1
- Python 3.x
Download videos from urls and put them in to a directory, such as:
python3 ./tools/video_download.py ./datasets/video_urls \
--video-dir ./datasets/video \
--video-type 'mp4' \
--threads 8 \Extract frames from video. Such as extract 1 frames per second and using 8 threads:
python3 ./tools/video2frame.py ./datasets/video_list \
--frame-dir ./datasets/frame \
--fps 1 \
--threads 8Extract features from videos, such as:
python3 main.py \
--video-dpath ./datasets/video \
--model resnet50 \
--batch-size 32 \
--save-dir ./datasets/feature