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)