作者:xh7212176 | 来源:互联网 | 2023-10-10 15:14
问题:训练神经网络,往往需要大量的标签和图像,标签名称和图像名称要求一一对应,数量也要相同,因为有的图像无合适的标签,所以可能会存在图像数量远大于标签数量的情况。
目的:编写代码,自动删除无标签的图像。
图1 问题截图
图2 运行成功结果图
代码示例:(含注释)
import os# 遍历 path 中所有文件的名称(绝对路径)
img_files = os.listdir(r"G:\CSDN\python\Office_skills\DDF\DDF_files/")i = 0 # 初始化索引
for img_file in img_files:# 遍历到倒数第二个文件if i
>>>output
保留的文件名称: raw-hat00-02-54.png
2
保留的文件名称: raw-hat00-04-23.png
4
删除的文件名称: raw-hat00-38-06.png
保留的文件名称: raw-hat00-42-32.png
6
保留的文件名称: raw-hat06-14-10.png
8
删除的文件名称: raw-hat06-14-21.png
保留的文件名称: raw-hat06-24-52.png
10
保留的文件名称: raw-hat06-41-14.png
12
保留的文件名称: raw-hat06-41-50.png
14
保留的文件名称: raw-hat07-16-36.png
16
保留的文件名称: raw-hat07-16-55.png
18
保留的文件名称: raw-hat07-24-31.png
20
保留的文件名称: raw-hat07-29-02.png
22
保留的文件名称: raw-hat07-53-24.png
24
保留的文件名称: raw-hat07-59-58.png
26
保留的文件名称: raw-hat08-00-09.png
28
保留的文件名称: raw-hat08-00-20.png
30
Process finished with exit code 0
>>>如有疑问,欢迎评论区一起探讨