(Python)脚本实现深度学习训练集合测试集的制作

代码可以正确运行,代码注释也已经添加

一、制作数据集实现图片和标签的对应

我们的需求是将图片制作成数据集,需要将不同类的图片放在不同的文件夹下面,然后生成一个 txt 文档,

文档内容是,每一行对应着图片名和该图片的标签。

比如这里我们指定微笑的图片对应的标签1,非微笑图片对应的标签为 0,已经将它们分文件夹存储好:
在这里插入图片描述
生成的效果:
在这里插入图片描述

# rootdir 就是个根目录
def listfiles(rootDir, txtfile

猜你喜欢

转载自blog.csdn.net/nanhuaibeian/article/details/106733639
今日推荐