Caltech-256 数据集处理(一) label提取

Caltech-256数据集在PyTorch中的处理:
Caltech-256 数据集处理(一) label提取
Caltech-256 数据集处理(二) 训练集和测试集的制作
Caltech-256 数据集处理(三) 训练集和验证集载入Dateloader

不知道为什么,最近Image-net的网站上不去了,花了好多功夫终于找到了暂且可以接受的分类替代数据集Caltech-256(https://authors.library.caltech.edu/7694/)
,其结构如下:
Caltech-256 数据集处理(一) label提取_第1张图片
Caltech-256 数据集处理(一) label提取_第2张图片
随手写了个脚本通过./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')

你可能感兴趣的:(python)