The code is referenced from: Yueqi Duan, Wenzhao Zheng, Xudong Lin, Jiwen Lu, and Jie Zhou, Deep Adversarial Metric Learning, Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition (CVPR). 2018: 2780-2789.b
Link for dataset. Download this before you run. Set the correct path in main.py
link to dataset
To train the model, run
python main.py