图像预处理完整案例 纯代码

__author__ = 'ding'
'''
完整的图像预处理案例
'''
import tensorflow as tf
import numpy as np
import matplotlib.pyplot as plt


def distort_color(image, color_ordering=0):
    if color_ordering == 0:
        image = tf.image.random_brightness(image, max_delta=32. / 255.)
        image = tf.image.random_saturation(image, lower=0.5, upper=1.5)
        image = tf.image.random_hue(image, max_delta=0.2)
        image = tf.image.random_contrast(image, lower=0.5, upper=1.5)
    elif color_ordering == 1:
        image = tf.image.random_saturation(image, lower=0.5, upper=1.5)
        image = tf.image.random_brightness(image, max_delta=32. / 255.)
        image = tf.image.random_contrast(image, lower=0.5, upper=1.5)
        image = tf.image.random_hue(image, max_delta=0.2)
    return tf.clip_by_value(image, 0.0, 1.0)


def preprocess_for_train(image, height, width, bbox):
    if bbox is None:
        bbox = tf.constant([0.0, 0.0, 1.0, 1.0], dtype=tf.float32, shape=[1, 1, 4])
    if image.dtype != tf.float32:
        image = tf.image.convert_image_dtype(image, dtype=tf.float32)

    bbox_begin, bbox_size, _ = tf.image.sample_distorted_bounding_box(
        tf.shape(image), bounding_boxes=bbox, min_object_covered=0.1
    )
    distort_image = tf.slice(image, bbox_begin, bbox_size)
    distort_image = tf.image.resize_images(
        distort_image, (height, width), method=np.random.randint(4)
    )
    distort_image = tf.image.random_flip_left_right(distort_image)
    distort_image = distort_color(distort_image, np.random.randint(2))
    return distort_image


image_raw_data = tf.gfile.GFile('./path/to/picture1.jpeg', 'rb').read()
with tf.Session() as sess:
    image_data = tf.image.decode_jpeg(image_raw_data)
    boxes = tf.constant([[[0.05, 0.05, 0.9, 0.7], [0.35, 0.47, 0.5, 0.56]]])

    for i in range(6):
        result = preprocess_for_train(image_data, 299, 299, boxes)
        plt.figure(i)
        plt.imshow(result.eval())
    plt.show()
  • 4
    点赞
  • 23
    收藏
    觉得还不错? 一键收藏
  • 4
    评论
蜡染图像是一种传统的染色技术,其色彩饱和度高、对比度强,但同时也存在着图像噪声和部分区域的失真问题。针对这些问题,可以使用OpenCV进行图像预处理来提高图像质量。 首先,可以使用高斯滤波器对图像进行平滑处理,减少图像中的噪声。代码如下: ```python import cv2 img = cv2.imread('wax_dye_image.jpg') blur_img = cv2.GaussianBlur(img, (5, 5), 0) cv2.imshow('blurred image', blur_img) cv2.waitKey(0) ``` 接下来,可以使用自适应阈值方法对图像进行二值化处理,以便更好地区分目标物体与背景。代码如下: ```python import cv2 img = cv2.imread('wax_dye_image.jpg') blur_img = cv2.GaussianBlur(img, (5, 5), 0) gray_img = cv2.cvtColor(blur_img, cv2.COLOR_BGR2GRAY) thresh_img = cv2.adaptiveThreshold(gray_img, 255, cv2.ADAPTIVE_THRESH_GAUSSIAN_C, cv2.THRESH_BINARY_INV, 11, 2) cv2.imshow('threshold image', thresh_img) cv2.waitKey(0) ``` 最后,可以使用形态学操作对图像进行进一步的处理,减少图像中的失真问题。代码如下: ```python import cv2 import numpy as np img = cv2.imread('wax_dye_image.jpg') blur_img = cv2.GaussianBlur(img, (5, 5), 0) gray_img = cv2.cvtColor(blur_img, cv2.COLOR_BGR2GRAY) thresh_img = cv2.adaptiveThreshold(gray_img, 255, cv2.ADAPTIVE_THRESH_GAUSSIAN_C, cv2.THRESH_BINARY_INV, 11, 2) kernel = np.ones((5, 5), np.uint8) closing_img = cv2.morphologyEx(thresh_img, cv2.MORPH_CLOSE, kernel) cv2.imshow('morphed image', closing_img) cv2.waitKey(0) ``` 通过这些处理,可以得到更加清晰、饱满的蜡染图像

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值