#图片数据集路径
import os
def generate(dir):
'''
dir:输入的文件路径
label:要生成的标签
'''
#返回dir中的所有文件,本例中是001.bad_apple和002.good_apple
files = os.listdir(dir)
#文件夹中的数据是按序排好的,否则可以先排序
#files.sort()
#要写入的文件名,若没有则生成一个新的
listText = open('images.txt','a')
for file in files:
#返回路径和文件名
fileType = os.path.split(file)
#print(file)
#print(fileType)
#图片数据集中可能会有一个readme是.txt格式的,忽略它
#readme文件也可能是其他格式的文件,根据修改即可
#若没有readme即可删掉
if fileType[1] == '.txt':
#print(fileType[1])
continue
#生成序号和标签
#i和label必须是同类型的
name = dir + file + '\n'
#生成文件名和标签
#name = file + ' ' + str(int(label)) +'\n'
listText.write(name)
listText.close()
out_path = 'E:\\data\\Benchmark\\Basketball\\img\\'
generate(out_path)
生成的images.txt放下c++同目录文件夹下。
并在该文件夹下新建文件:region.txt
里面的内容由数据集中groundtruth_rect.txt第一行数据计算而来:
我的是:198,214,198,295,232,214,232,295
参考:https://blog.csdn.net/happyday_d/article/details/84899341