├── README.md ├── assests └── XModalViT_Model.png ├── environment.yml ├── run_expt.sh └── src ├── datasets ├── __pycache__ │ ├── chair.cpython-38.pyc │ ├── shoe.cpython-38.pyc │ └── sketchy.cpython-38.pyc ├── chair.py ├── shoe.py └── sketchy.py ├── networks ├── __pycache__ │ ├── module.cpython-38.pyc │ ├── photo_encoder.cpython-38.pyc │ ├── photo_encoder_student.cpython-38.pyc │ ├── sketch_encoder.cpython-38.pyc │ ├── sketch_encoder_student.cpython-38.pyc │ └── teacher.cpython-38.pyc ├── module.py ├── photo_encoder.py ├── photo_encoder_student.py ├── sketch_encoder.py ├── sketch_encoder_student.py └── teacher.py ├── params ├── __pycache__ │ ├── chair.cpython-38.pyc │ ├── shoe.cpython-38.pyc │ └── sketchy.cpython-38.pyc ├── chair.py ├── shoe.py └── sketchy.py ├── test.py ├── train_photo_student.py ├── train_sketch_student.py ├── train_teacher.py └── utils ├── __pycache__ └── losses.cpython-38.pyc └── losses.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhrac/xmodal-vit/HEAD/README.md -------------------------------------------------------------------------------- /assests/XModalViT_Model.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhrac/xmodal-vit/HEAD/assests/XModalViT_Model.png -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhrac/xmodal-vit/HEAD/environment.yml -------------------------------------------------------------------------------- /run_expt.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhrac/xmodal-vit/HEAD/run_expt.sh -------------------------------------------------------------------------------- /src/datasets/__pycache__/chair.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhrac/xmodal-vit/HEAD/src/datasets/__pycache__/chair.cpython-38.pyc -------------------------------------------------------------------------------- /src/datasets/__pycache__/shoe.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhrac/xmodal-vit/HEAD/src/datasets/__pycache__/shoe.cpython-38.pyc -------------------------------------------------------------------------------- /src/datasets/__pycache__/sketchy.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhrac/xmodal-vit/HEAD/src/datasets/__pycache__/sketchy.cpython-38.pyc -------------------------------------------------------------------------------- /src/datasets/chair.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhrac/xmodal-vit/HEAD/src/datasets/chair.py -------------------------------------------------------------------------------- /src/datasets/shoe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhrac/xmodal-vit/HEAD/src/datasets/shoe.py -------------------------------------------------------------------------------- /src/datasets/sketchy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhrac/xmodal-vit/HEAD/src/datasets/sketchy.py -------------------------------------------------------------------------------- /src/networks/__pycache__/module.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhrac/xmodal-vit/HEAD/src/networks/__pycache__/module.cpython-38.pyc -------------------------------------------------------------------------------- /src/networks/__pycache__/photo_encoder.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhrac/xmodal-vit/HEAD/src/networks/__pycache__/photo_encoder.cpython-38.pyc -------------------------------------------------------------------------------- /src/networks/__pycache__/photo_encoder_student.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhrac/xmodal-vit/HEAD/src/networks/__pycache__/photo_encoder_student.cpython-38.pyc -------------------------------------------------------------------------------- /src/networks/__pycache__/sketch_encoder.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhrac/xmodal-vit/HEAD/src/networks/__pycache__/sketch_encoder.cpython-38.pyc -------------------------------------------------------------------------------- /src/networks/__pycache__/sketch_encoder_student.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhrac/xmodal-vit/HEAD/src/networks/__pycache__/sketch_encoder_student.cpython-38.pyc -------------------------------------------------------------------------------- /src/networks/__pycache__/teacher.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhrac/xmodal-vit/HEAD/src/networks/__pycache__/teacher.cpython-38.pyc -------------------------------------------------------------------------------- /src/networks/module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhrac/xmodal-vit/HEAD/src/networks/module.py -------------------------------------------------------------------------------- /src/networks/photo_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhrac/xmodal-vit/HEAD/src/networks/photo_encoder.py -------------------------------------------------------------------------------- /src/networks/photo_encoder_student.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhrac/xmodal-vit/HEAD/src/networks/photo_encoder_student.py -------------------------------------------------------------------------------- /src/networks/sketch_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhrac/xmodal-vit/HEAD/src/networks/sketch_encoder.py -------------------------------------------------------------------------------- /src/networks/sketch_encoder_student.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhrac/xmodal-vit/HEAD/src/networks/sketch_encoder_student.py -------------------------------------------------------------------------------- /src/networks/teacher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhrac/xmodal-vit/HEAD/src/networks/teacher.py -------------------------------------------------------------------------------- /src/params/__pycache__/chair.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhrac/xmodal-vit/HEAD/src/params/__pycache__/chair.cpython-38.pyc -------------------------------------------------------------------------------- /src/params/__pycache__/shoe.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhrac/xmodal-vit/HEAD/src/params/__pycache__/shoe.cpython-38.pyc -------------------------------------------------------------------------------- /src/params/__pycache__/sketchy.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhrac/xmodal-vit/HEAD/src/params/__pycache__/sketchy.cpython-38.pyc -------------------------------------------------------------------------------- /src/params/chair.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhrac/xmodal-vit/HEAD/src/params/chair.py -------------------------------------------------------------------------------- /src/params/shoe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhrac/xmodal-vit/HEAD/src/params/shoe.py -------------------------------------------------------------------------------- /src/params/sketchy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhrac/xmodal-vit/HEAD/src/params/sketchy.py -------------------------------------------------------------------------------- /src/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhrac/xmodal-vit/HEAD/src/test.py -------------------------------------------------------------------------------- /src/train_photo_student.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhrac/xmodal-vit/HEAD/src/train_photo_student.py -------------------------------------------------------------------------------- /src/train_sketch_student.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhrac/xmodal-vit/HEAD/src/train_sketch_student.py -------------------------------------------------------------------------------- /src/train_teacher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhrac/xmodal-vit/HEAD/src/train_teacher.py -------------------------------------------------------------------------------- /src/utils/__pycache__/losses.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhrac/xmodal-vit/HEAD/src/utils/__pycache__/losses.cpython-38.pyc -------------------------------------------------------------------------------- /src/utils/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhrac/xmodal-vit/HEAD/src/utils/losses.py --------------------------------------------------------------------------------