学习总结笔记

xiaoxiao2021-02-28  91

一、数据集准备

利用该命令行,读取指定路径下图像文件的路径保存到文本文件中,用于制作训练集

ls -R /dir/*.jpg > file.txt

也可以通过一段python代码生成,如下所示:

import os root = os.getcwd() #获取当前路径 data = 'images' path = os.listdir(root+'/'+ data) #显示该路径下所有文件 path.sort() file = open('train.txt','w') i = 0 for line in path: str = root+'/'+ data +'/'+line for child in os.listdir(str): str1 = data+'/'+line+'/'+child; d = ' %s' %(i) t = str1 + d file.write(t +'\n') i=i+1 file.close()

转载请注明原文地址: https://www.6miu.com/read-96989.html

最新回复(0)