提取文件中的每一个mask,并将mask命名为文件名字

import cv2 as cv
import random
import glob
import os
from PIL import Image
import shutil


def get_samples(foldername, savePath):
    print('savePath:', savePath)
    if os.path.exists(savePath) is False:
        os.makedirs(savePath)

    filenames = os.listdir(foldername)

    for filename in filenames:
        full_path = os.path.join(foldername, filename)
        new_name = filename[:-5] + '.png'
        label_png = os.listdir(full_path)[2]
        # os.rename(os.path.join(filename, label_png),os.path.join(filename, name))
        shutil.copy(os.path.join(full_path, label_png), os.path.join(savePath, label_png))
        os.rename(os.path.join(savePath, label_png), os.path.join(savePath, new_name))
        # print(os.listdir(filename))

savePath = 'C:/Users/Desktop/978/'
get_samples('C:/Users/Desktop/978/', savePath)

猜你喜欢

转载自www.cnblogs.com/apan008/p/11563232.html