删除文件夹中的相邻但名称不同的文件 && 适应于神经网络训练时的标签名称和图像名称相对应的情况

最新推荐文章于 2022-12-09 13:36:59 发布

Flying Bulldog

最新推荐文章于 2022-12-09 13:36:59 发布

阅读量193

点赞数

分类专栏： python_skills

本文链接：https://blog.csdn.net/qq_54185421/article/details/124856180

版权

神经网络深度学习机器学习 python

python_skills 专栏收录该内容

8 篇文章 1 订阅

订阅专栏

问题：训练神经网络，往往需要大量的标签和图像，标签名称和图像名称要求一一对应，数量也要相同，因为有的图像无合适的标签，所以可能会存在图像数量远大于标签数量的情况。

目的：编写代码，自动删除无标签的图像。

图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 < len(img_files):
        
        # 判断 第一个件的名称 和 第二个文件的名称 是否一样
        if img_files[i].split('.')[0] != img_files[i+1].split('.')[0]:
            
            # 如果不一样，删除第一个文件
            os.remove(r"G:\CSDN\python\Office_skills\DDF\DDF_files/{}".format(img_files[i]))
            print('删除的文件名称：', img_files[i])
            
            # 删除文件之后，重新读取文件夹中的文件，更新所有文件的数量
            img_files = os.listdir(r"G:\CSDN\python\Office_skills\DDF\DDF_files/")
            
        # 如果一样，索引+2
        else:
            print('保留的文件名称：', img_files[i])
            i += 2
            print(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