metric learning divide and conquer:CVPR 2019论文“划分和征服度量学习的嵌入空间”的源代码 Source code l
划分和征服度量学习的嵌入空间 关于 该存储库包含用于复制数据集 , 和的结果以(CVPR 2019)的代码。 纸: 补充文件: 我们还将我们的方法应用于Kaggle的,在2131名中排名第10位。 幻灯片: 要求 Python版本3.6.6或更高版本 SciPy和scikit-learn软件包 PyTorch( ) 支持GPU的 ( ) 下载并提取,和的数据集 用法 以下命令将在200个纪元和80个批次的店内衣服数据集上训练具有边际损失的模型,同时将嵌入层分为8个簇,并从190个纪元开始对模型进行微调。 通过简单地将--dataset=inshop为--dataset=sop (Stanford Online Products)或--dataset=vid (Vehicle-ID),可以使用此命令为三个数据集重现论文结果。 CUDA_VISIBLE_DEVICES=0 p
下载地址
用户评论