代码参考:https://github.com/msight-tech/research-ms-loss
论文参考:Wang_Multi-Similarity_Loss_With_General_Pair_Weighting_for_Deep_Metric_Learning_CVPR_2019_paper.pdf
1.环境准备
ubuntu16.04
cuda10.1
torch==1.7.1
torchvision==0.8.2
numpy==1.19.5
yacs==0.1.8
Pillow==8.1.0
cd reseach-ms-loss
python setup.py develop build
2.准备预训练模型
2) resnet5