-
torch
-
cv2
-
librosa
-
numpy
-
pydub
conda install numpy
conda install -c pytorch pytorch
conda install -c conda-forge opencv librosa pydub
python video_gen.py -i videos -o frames
python audio_gen.py -i videos -o spectrograms
python model.py -f frames -s spectrograms
Loophole: Transforms passed into DataLoaders (as argument) might not work