pytorch学习（二）—自定义数据集

#在深度学习中常常须要生成带标签的图片名称列表，xxxlist.txt文件， #编写脚本语言，实现对文件中图片生成带标签的txt文件方法 import os def generate(dir,label): files = os.listdir(dir) files.sort() print("*****************") print("input =

>>阅读原文<<