如何训练自己的数据集？

ZF4444 / MMAL-Net

This is a PyTorch implementation of the paper "Multi-branch and Multi-scale Attention Learning for Fine-Grained Visual Categorization (MMAL-Net)" (Fan Zhang, Meng Li, Guisheng Zhai, Yizhao Liu).

250 stars 57 forks source link

如何训练自己的数据集？ #38

Closed WangRongsheng closed 2 years ago

WangRongsheng commented 2 years ago

您好，目前我正在参加一项医学图像的分类竞赛，我发现该竞赛中所提供的数据适用细粒度分类，通过前期学习调研考虑应用您的模型，但是我并没有找到相关训练自己数据的方法？我的数据集是采用文件夹名称进行划分的：

dataset
|
|--类别1
|----1.jpg
|----...
|--类别2
|----1.jpg
|----...
|--类别3
|----1.jpg
|----...

请问我该如何重新组织它以便放入模型进行训练？