leaderj1001 / MobileNetV3-Pytorch

Implementing Searching for MobileNetV3 paper using Pytorch
MIT License
292 stars 70 forks source link

数据集制作 #14

Open Galaxy-yyyyy opened 4 years ago

Galaxy-yyyyy commented 4 years ago

请问如果用自己的数据训练大佬写的程序,数据集怎么制作了,1,不用再另外制作label,文件夹就是标签,有多少个文件夹就有多少类别;2,需要另外制作标签;

AishuaiYao commented 4 years ago

请问如果用自己的数据训练大佬写的程序,数据集怎么制作了,1,不用再另外制作label,文件夹就是标签,有多少个文件夹就有多少类别;2,需要另外制作标签;

from torchvision import datasets,transforms

datasets的ImageFolder库自己百度下