Features are saved as doubles

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

Open v-iashin opened 9 months ago

v-iashin commented 9 months ago

np.array(vid_feats) saves the list into float64 which doubles the disk space.

casting it to f32 (or, ideally, to a user-defined precision) would be nice.

it is should be very simply to do