对图像进行裁剪后，删掉没有前景目标的图像

最新推荐文章于 2025-12-07 20:47:24 发布

原创最新推荐文章于 2025-12-07 20:47:24 发布 · 149 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#人工智能 #深度学习

code 专栏收录该内容

3 篇文章

订阅专栏

该文章介绍了一个Python函数，用于检查指定路径下的空文件，并根据给定的标签路径移除相应的图片。主要涉及os和path模块的使用。

from os.path import isdir, abspath, getsize, join
from os import listdir, system, remove
import os

 
def get_del_name(label_path, get_remove=False):
    del_names = []
    file_names = listdir(abspath(label_path))
    for file_name in file_names:
        content = join(label_path, file_name)
        if getsize(content) == 0:
            name = file_name.split(".")[0]
            del_names.append(name)
            if get_remove:
                remove(content)

    return del_names


def del_images(img_path, del_names):
    for name in del_names:
        image_name = name + ".png"
        content = join(img_path, image_name)
        if os.path.isfile(content):
            remove(content)
               

def main():
    
    label_path = "/datasets/val/label/"
    img_path = "/datasets/val/images/"
    
    del_names = get_del_name(label_path, True)
    del_images(img_path, del_names)
    
 
if __name__ == '__main__':
    main()