image_set.py 文件源码

def save_gray(src_dir, dst_dir, size):
    for full_filename in glob.iglob(src_dir + '*.png'):
        image = cv2.imread(full_filename)
        gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
        small = cv2.resize(gray, (size, size), interpolation=cv2.INTER_AREA)
        new_filename = full_filename.replace(src_dir, dst_dir)
        cv2.imwrite(new_filename, small)