weixin_43955398的博客

原创 【pytorch】创建自己的数据集，并按类别的比例划分得到均匀的数据集

会导致每一类的数据取得不均匀，也就是说，可能类别0取了30个，类别1取了26个，类别2取了38个这样的情况，我们希望每个类别都取30个。加载数据集使用datasets中的ImageFolder函数即可快速加载上图形式的数据。，共900条数据，如图-1所示，每个文件夹中包含一类数据，共6个文件夹。划分数据集，但又不想人为移动文件进行手动划分。数据集有6个类别，每个类别的数据有150个。