Caltech-256数据集在PyTorch中的处理:
Caltech-256 数据集处理(一) label提取
Caltech-256 数据集处理(二) 训练集和测试集的制作
Caltech-256 数据集处理(三) 训练集和验证集载入Dateloader
不知道为什么,最近Image-net的网站上不去了,花了好多功夫终于找到了暂且可以接受的分类替代数据集Caltech-256(https://authors.library.caltech.edu/7694/)
,其结构如下:
随手写了个脚本通过./256_ObjectCategories/下的文件夹生成label.txt。
import os
datasetpath="/media/this/02ff0572-4aa8-47c6-975d-16c3b8062013/256_ObjectCategories"
d=os.listdir(datasetpath)
d.sort()
with open(r'label.txt','w',encoding='utf-8') as f:
for i in d:
f.write(i)
f.write('\n')