0
点赞
收藏
分享

微信扫一扫

根据文件名匹配,在png中找出对应的jpg

陈情雅雅 2022-02-11 阅读 85
python

jpg文件比较多,需要找出png文件

from glob import glob
import os
from shutil import copyfile
from tqdm import tqdm

if __name__ == '__main__':
    segdir = "SegmentationClass/"
    jpgdir = "JPEGImages0/"
    

    # now we are going to find jpg matches seg
    seglist = os.listdir(segdir)
    print(f'len(seglist)={len(seglist)}')
    # for filename in seglist:
    #     print(filename)

    jpglist = os.listdir(jpgdir)
    print(f'len(jpglist)={len(jpglist)}')
    jpgsavedir = "JPEGImages/"
    if not os.path.exists(jpgsavedir):
        os.mkdir(jpgsavedir)
    
    for jpgname in tqdm(jpglist):
        if (jpgname[:-4]+".png") in seglist:
            copyfile(jpgdir+jpgname,jpgsavedir+jpgname)

举报

相关推荐

0 条评论