search

v-iashin / video_features

Extract video features from raw videos using multiple GPUs. We support RAFT flow frames as well as S3D, I3D, R(2+1)D, VGGish, CLIP, and TIMM models.
https://v-iashin.github.io/video_features
MIT License
532 stars 97 forks source link

Feature Extraction Problem in RTX 3090 #100

Closed rabeya-akter closed 1 year ago

rabeya-akter commented 1 year ago

I am trying to extract feature_type = i3d in RTX 3090. But it freezes the terminal after printing this:

feature_type: i3d stack_size: 12 step_size: 2 streams: rgb flow_type: pwc extraction_fps: null device: cuda:0 on_extraction: print output_path: ./output/i3d tmp_path: ./tmp/i3d keep_tmp_files: false show_pred: false config: null video_paths: null file_with_video_paths: ./sample/sample_video_paths.txt

Device: cuda:0

and doesn't do anything after this.

v-iashin commented 1 year ago

duplicate of #13