生成训练路径 .txt 地址文件

生成训练地址路径 .txt 文件

读取两个文件目录下的地址，然后合并输出到txt文件中

def create_txt(in_image,in_label,out_txt):
    # 查看in_image,in_label目录下的图片
    images = os.listdir(in_image)
    labels = os.listdir(in_label)
    
    # 位置匹配
    images = sorted(images)
    labels = sorted(labels)
    
    print(images[0])
    print(labels[0])

    if not os.path.exists(out_txt):
        # 创建txt文件
        os.mknod(out_txt)
        print('txt创建成功：',out_txt)
    
    # 打开txt文件
    f = open(out_txt,'w')
    print('txt打开成功：',out_txt)
    # 遍历所有路径，开始写入
    print('开始写入：')
    for i in range(len(images)):
        img_path = os.path.join(in_image,images[i])
        label_path = os.path.join(in_label,labels[i])
        path = img_path + ' ' + label_path
        f.write("{}\n".format(path))
    
    f.close()
    print('写入完成:',out_txt)