构建图像分类数据集

  1. 获取原始图片

  1. 利用网络爬虫爬取

  1. 自己拍摄(一般不推荐)

  1. 一些注意事项

  1. 手动删除一些不符的图像(针对网络爬虫方式)

  1. 各类别要尽量均匀

  1. 对于每一类别的图像,我们要保证其多样性,以避免过拟合

  1. 存储数据集

同一类图像放在同一文件夹下

  1. 划分训练集与测试集

将训练集放在train目录下

测试集放在val目录下

参考

github.com/TommyZihao/Train_Custom_Dataset

同济子豪兄

你可能感兴趣的:(python,深度学习)