Refactoring inference.py

leeesangwon / PyTorch-Image-Retrieval

A PyTorch framework for an image retrieval task including implementation of N-pair Loss (NIPS 2016) and Angular Loss (ICCV 2017).

MIT License

275 stars 55 forks source link

Closed MikeKook closed 5 years ago

MikeKook commented 5 years ago

Refactoring inference.py, remove flip-test option on main.py and data_loader.py