├── .gitignore ├── LICENSE ├── README.md ├── assets ├── ex1.jpeg ├── ex2.jpeg └── training_hyperparameters.png ├── colab ├── eval.ipynb ├── train_T5_Base.ipynb └── train_T5_Large.ipynb ├── env.yml ├── eval.py ├── modules ├── __init__.py ├── multi_frame_dataset.py └── multi_frame_model.py └── train.py /.gitignore: -------------------------------------------------------------------------------- 1 | /data/ 2 | /multi_frame_results/ 3 | -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akshaygopalkr/EM-VLM4AD/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akshaygopalkr/EM-VLM4AD/HEAD/README.md -------------------------------------------------------------------------------- /assets/ex1.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akshaygopalkr/EM-VLM4AD/HEAD/assets/ex1.jpeg -------------------------------------------------------------------------------- /assets/ex2.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akshaygopalkr/EM-VLM4AD/HEAD/assets/ex2.jpeg -------------------------------------------------------------------------------- /assets/training_hyperparameters.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akshaygopalkr/EM-VLM4AD/HEAD/assets/training_hyperparameters.png -------------------------------------------------------------------------------- /colab/eval.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akshaygopalkr/EM-VLM4AD/HEAD/colab/eval.ipynb -------------------------------------------------------------------------------- /colab/train_T5_Base.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akshaygopalkr/EM-VLM4AD/HEAD/colab/train_T5_Base.ipynb -------------------------------------------------------------------------------- /colab/train_T5_Large.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akshaygopalkr/EM-VLM4AD/HEAD/colab/train_T5_Large.ipynb -------------------------------------------------------------------------------- /env.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akshaygopalkr/EM-VLM4AD/HEAD/env.yml -------------------------------------------------------------------------------- /eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akshaygopalkr/EM-VLM4AD/HEAD/eval.py -------------------------------------------------------------------------------- /modules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /modules/multi_frame_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akshaygopalkr/EM-VLM4AD/HEAD/modules/multi_frame_dataset.py -------------------------------------------------------------------------------- /modules/multi_frame_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akshaygopalkr/EM-VLM4AD/HEAD/modules/multi_frame_model.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akshaygopalkr/EM-VLM4AD/HEAD/train.py --------------------------------------------------------------------------------