数据增强——网格形变

最新推荐文章于 2025-08-15 17:11:38 发布

原创

最新推荐文章于 2025-08-15 17:11:38 发布 · 633 阅读

3 ·

CC 4.0 BY-SA版权

本文介绍如何在牙片根管治疗图像中应用网格形变进行数据增强，以保留图像细节，提升处理效果。

数据增强——网格形变

牙片根管治疗需要保存图像细节，适合使用网格形变进行数据扩充

效果图：

代码：

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")




# taken from: https://www.kaggle.com/bguberfain/elastic-transform-for-data-augmentation
# Function to distort image
def elastic_transform(image, alpha, sigma, alpha_affine, random_state