找到后,将后缀改为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) # 看一下列表里的内容