
PMG-Progressive-Multi-Granularity-Training:利用渐进式拼图补丁进行多粒度训练以改善细粒度分类...
5星
- 浏览量: 0
- 大小:None
- 文件类型:ZIP
简介:
PMG-Progressive-Multi-Granularity-Training是一种创新性的深度学习方法,通过采用渐进式的拼图补丁技术进行多粒度训练,显著提升了图像的细粒度分类精度。
渐进式多粒度训练
使用通过拼图补丁的渐进式方法进行视觉精细分类任务的代码已在ECCV2020上发布。
所需环境:
- Python 3.6
- PyTorch >= 1.3.1
- Torchvision >= 0.4.2
训练步骤:
下载FGVC的数据集(如CUB-200-2011,Stanford Cars,FGVC-Aircraft等),然后按照以下结构组织数据:
```
dataset
├── train
│ ├── class_001
| | ├── 1.jpg
| | ├── 2.jpg
| | └── ...
│ ├── class_002
| | ├── 1.jpg
| | ├── 2.jpg
| | └── ...
│ └── ...
└── test
├── class_001
│ ├─...
```
请根据上述结构准备数据集,以便顺利进行训练。
全部评论 (0)
还没有任何评论哟~


