c++版KCF调试运行时,需要添加的images.txt文件生成代码

#图片数据集路径
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


版权声明:本文为u010440456原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。