项目作者: Tramac

项目描述 :
Video feature extractor in PyTorch.
高级语言: Python
项目地址: git://github.com/Tramac/awesome-video-feature-extractor.git
创建时间: 2020-07-28T03:51:29Z
项目社区:https://github.com/Tramac/awesome-video-feature-extractor

开源协议:Apache License 2.0

下载


awesome-video-feature-extractor

python-image
pytorch-image
lic-image

Video feature extracor for your own datasets.

Environment

  • Ubuntu 16.04
  • CUDA 9.0
  • cuDNN 7.5.1
  • Python 3.x

Usage

Video DownLoad (optional)

Download videos from urls and put them in to a directory, such as:

  1. python3 ./tools/video_download.py ./datasets/video_urls \
  2. --video-dir ./datasets/video \
  3. --video-type 'mp4' \
  4. --threads 8 \

Video CutFrame (optional)

Extract frames from video. Such as extract 1 frames per second and using 8 threads:

  1. python3 ./tools/video2frame.py ./datasets/video_list \
  2. --frame-dir ./datasets/frame \
  3. --fps 1 \
  4. --threads 8

Extract Features

Extract features from videos, such as:

  1. python3 main.py \
  2. --video-dpath ./datasets/video \
  3. --model resnet50 \
  4. --batch-size 32 \
  5. --save-dir ./datasets/feature

References