MobileNetV3-Pytorch icon indicating copy to clipboard operation
MobileNetV3-Pytorch copied to clipboard

数据集制作

Open Galaxy-yyyyy opened this issue 5 years ago • 1 comments

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

Galaxy-yyyyy avatar Jan 02 '20 03:01 Galaxy-yyyyy

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

from torchvision import datasets,transforms

datasets的ImageFolder库自己百度下

AishuaiYao avatar Mar 01 '20 02:03 AishuaiYao