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
2021-12-15 18:22:57
881KB
Python
1