moabitcoin / ig65m-pytorch

PyTorch 3D video classification models pre-trained on 65 million Instagram videos
MIT License
265 stars 30 forks source link

about extract features #41

Closed kimsimple closed 3 years ago

kimsimple commented 3 years ago

Thank you for your wanderful work!

Example for running on GPUs via nvidia-docker:

docker run --runtime=nvidia --ipc=host -v $PWD:/data moabitcoin/ig65m-pytorch:latest-gpu \
    extract /data/myvideo.mp4 /data/myfeatures.npy

Could this docker extract one batch feature one time? 😥

kimsimple commented 3 years ago

This problem was solved: https://github.com/moabitcoin/ig65m-pytorch/issues/6 😀