PyTorch implementation for multimodal learning with discrete representations.
git clone [your-repo-url]
cd src_cvpr
# Create environment with Python 3.10.16
conda create -n codaar python=3.10.16
conda activate codar
# Install requirements
pip install -r requirements.txtAudio-Visual (AV) Setting:
cd src_cvpr
sbatch pretrain_novel_AV.sbatchAudio-Visual-Text (AVT) Setting:
cd src_cvpr
sbatch pretrain_novel_AVT.sbatchAVE:
cd src_cvpr
sbatch ave_novel.sbatchAVVP:
cd src_cvpr
sbatch avvp_novel.sbatchAVE→AVVP:
cd src_cvpr
sbatch ave_avvp_novel.sbatchUCF-VGGSound:
cd src_cvpr
sbatch ucf_vggsound_novel.sbatchAVS:
# Training
cd src_cvpr/AVSBench_downstream/avs_scripts/avs_s4
sbatch train_novel.sbatch
# Testing
cd src_cvpr/AVSBench_downstream/avs_scripts/avs_s4
sbatch test_novel.sbatchDataset CSV files are located in src_cvpr/data/
- VGGSound90k: https://github.com/hche11/VGGSound
- VGGSound40k: https://github.com/jasongief/CPSP
- AVE: https://github.com/YapengTian/AVE-ECCV18
- AVVP: https://github.com/YapengTian/AVVP-ECCV20
- AVSBench-S4: https://github.com/OpenNLPLab/AVSBench
- UCF-101: https://www.crcv.ucf.edu/data/UCF101.php
For video and audio feature extraction methods, please refer to AVE.