Smooth_AP:ECCV '20论文“ Smooth AP”的代码 源码

Smooth_AP ECCV '20论文 src / Smooth_AP_loss.py中提供了Smooth-AP损失功能的PyTorch实现 训练代码和预先训练的重量即将推出... 依存关系 的Python 3.7.7 PyTorch 1.6.0 CUDA 10.1 数据 此存储库用于在以下数据集上使用Smooth-AP损失进行训练: PKU车辆ID(可从该网站获得-必须通过电子邮件发送给作者以获取下载权限) INaturalist(2018版本-从此网站获得) 我们是第一个将大型INaturalist数据集用于图像复原任务的人。数据集拆分可在此处下载: : 。将压缩包解压缩到INaturalist数据集目录中。 训练模型 可以使用此存储库复制“车辆ID”和“自然”数据集的训练结果。要在Vehicle ID数据集上训练模型,可以运行: python main.py --f