multi_augmentation_pic.py 文件源码

def mirror_pic(output_data_path, _dir, pic, img):

    fname, fextension = os.path.splitext(pic)

    mirror_x_img = img[:, ::-1, :]
    mirror_x_img_gray = rgb2gray(mirror_x_img)
    mirror_y_img = img[::-1, :, :]
    mirror_y_img_gray = rgb2gray(mirror_y_img)
    mirror_xy_img = img[::-1, ::-1, :]
    mirror_xy_img_gray = rgb2gray(mirror_xy_img)

    misc.imsave(os.path.join(output_data_path, _dir, (fname + '_mirror_x' + fextension)), mirror_x_img_gray)
    os.chmod(os.path.join(output_data_path, _dir, (fname + '_mirror_x' + fextension)), stat.S_IWRITE)
    misc.imsave(os.path.join(output_data_path, _dir, (fname + '_mirror_y' + fextension)), mirror_y_img_gray)
    os.chmod(os.path.join(output_data_path, _dir, (fname + '_mirror_y' + fextension)), stat.S_IWRITE)
    misc.imsave(os.path.join(output_data_path, _dir, (fname + '_mirror_xy' + fextension)), mirror_xy_img_gray)
    os.chmod(os.path.join(output_data_path, _dir, (fname + '_mirror_xy' + fextension)), stat.S_IWRITE)

    return mirror_x_img, mirror_y_img, mirror_xy_img