DUTS数据集中DUTS\DUTS-TR\DUTS-TR-Mask中的命名错误的图片名字

1
找到后,将后缀改为png,替换掉之前存在的同名png图片即可。
找出代码:

import os
 
f = open("./img.txt", 'w')  # 先创建一个空的文本 img
path = "./DUTS-TR-Image/"  # 指定需要读取文件的目录 DUTS-TR-Mask DUTS-TR-Image
files = os.listdir(path)  # 采用listdir来读取所有文件
files.sort()  # 排序
s = []  # 创建一个空列表
num = 0
for file_ in files:  # 循环读取每个文件名
    num = num + 1
    #    print(path +file_)
    if not os.path.isdir(path + file_):  # 判断该文件是否是一个文件夹
        f_name = str(file_)
        #        print(f_name)
        s.append(f_name)  # 把当前文件名返加到列表里
        f.write(str(num) + ' ' + f_name + '\n')  # 写入之前的文本中
 
print(s)  # 看一下列表里的内容