图像分类：批量建立dataset数据集的分类.txt类型标签

十二口

已于 2023-05-16 19:00:51 修改

阅读量225

点赞数

分类专栏： pytorch 文章标签：分类 python pytorch

于 2023-05-16 17:10:05 首次发布

本文链接：https://blog.csdn.net/qq_41940277/article/details/130709738

版权

pytorch 专栏收录该内容

8 篇文章 0 订阅

订阅专栏

图像分类：批量建立dataset数据集的分类.txt类型标签

https://www.bilibili.com/video/BV1hE411t7RN?p=7&spm_id_from=pageDriver&vd_source=5b6e0605c1ed0f1db9c92503dd5994e0

import os
root_dir="dataset/train"
target_dir="ants_image"
img_path=os.listdir(os.path.join(root_dir,target_dir)
label=target_dir.split('_')[0]
out_dir="ant_label"
for i in img_path:
	file_name=i.split('.jpg')[0]
	with open(os.path.join(root_dir,out_dir,"{}.txt=".format(file_name)),'w') as f:
		f.write(label)