数据增强——网格形变
牙片根管治疗需要保存图像细节,适合使用网格形变进行数据扩充
效果图:
代码:
import numpy as np # linear algebra
import pandas as pd # data processing, CSV file I/O (e.g. pd.read_csv)
import skimage.io
from skimage import color
from skimage import io
import glob
import cv2
from scipy.ndimage.interpolation import map_coordinates
from scipy.ndimage.filters import gaussian_filter
import matplotlib.pyplot as plt
import scipy.misc
from PIL import Image
image_mask_dir = "C/label/"
all_masks_files = glob.glob(image_mask_dir+"*.jpg")
image_dir = "C/image/"
all_image_files = glob.glob(image_dir+"*.jpg