TensorFlow图像处理API
2021-04-11 16:26
标签：用户 ima 深度学习 实践 class oat span lib color TensorFlow提供了一些常用的图像处理接口，可以让我们方便的对图像数据进行操作，以下首先给出一段显示原始图片的代码，然后在此基础上，实践TensorFlow的不同API。 运行效果如下图： 其中tf.gfile.FastGFile，用于读取本地文件，tf.image.decode_jpeg用于将jpeg图片原始数据解码到3-D张量空间，即width, height, channel，最后调用pyplt库显示图片。 TensorFlow缩放图片的接口是tf.image.resize_images，[200， 200]是缩放后的目标尺寸，这里调用了tf.cast这个类型转换函数，因为经过缩放处理后，张量类型为float32，而pyplt对图像格式要求uint8，所以必须转一下，否则什么效果，可以自己试一下。 tf.image.resize_images接口可以指定不同的缩放算法，比如： tf.image.resize_images(img_data, [200, 200], method=tf.image.ResizeMethod.BICUBIC) 图像的反转在各路深度学习算法中就用的比较多了，主要是通过这种操作可以扩大样本的数量，何乐不为。 上述代码调用了左右反转接口，TensorFlow还提供了上下反转及随机反转的操作，不再一一尝试。 上述代码指定左上角的200px方形box进行裁剪，指定目标范围必须合理，否则会产生异常。 这段代码有几个地方要注意一下，在jpeg解码后，调用了tf.expand_dims，这个函数的意思是在指定的位置增加一个维度，因为解码后是3维数据，在0位置增加一维，事实上增加了一个batch维度，如此操作主要是为了迎合后面的画框函数！boxes操作节点定义了两个方框，用0~1的浮点数标识box的位置比例，最后的图片显示位置也要注意，输出是四维，请取出第一个图片显示。下图为显示效果，手工放大图片后的效果，否则，1px方框在plt中可能被缩略掉，请注意！ TensorFlow图像处理API 标签：用户 ima 深度学习 实践 class oat span lib color 原文地址：https://www.cnblogs.com/billux/p/9024720.html显示原始图片
1 import matplotlib.pyplot as plt
2 import tensorflow as tf
3
4 raw_data = tf.gfile.FastGFile(‘./new.jpg‘,‘rb‘).read()
5
6 with tf.Session() as sess:
7 img_data = tf.image.decode_jpeg(raw_data)
8 plt.imshow(img_data.eval())
9 plt.show()
图像的缩放
1 import matplotlib.pyplot as plt
2 import tensorflow as tf
3
4 raw_data = tf.gfile.FastGFile(‘./new.jpg‘,‘rb‘).read()
5
6 with tf.Session() as sess:
7 img_data = tf.image.decode_jpeg(raw_data)
8 img_data2 = tf.cast(tf.image.resize_images(img_data, [200, 200]), dtype=tf.uint8);
9 plt.imshow(img_data2.eval())
10 plt.show()
图像的反转
1 import matplotlib.pyplot as plt
2 import tensorflow as tf
3
4 raw_data = tf.gfile.FastGFile(‘./new.jpg‘,‘rb‘).read()
5
6 with tf.Session() as sess:
7 img_data = tf.image.decode_jpeg(raw_data)
8 img_data2 = tf.cast(tf.image.flip_left_right(img_data), dtype=tf.uint8)
9 plt.imshow(img_data2.eval())
10 plt.show()
图像的裁剪
中心裁剪
1 import matplotlib.pyplot as plt
2 import tensorflow as tf
3
4 raw_data = tf.gfile.FastGFile(‘./new.jpg‘,‘rb‘).read()
5
6 with tf.Session() as sess:
7 img_data = tf.image.decode_jpeg(raw_data)
8 img_data2 = tf.cast(tf.image.resize_image_with_crop_or_pad(img_data, 200, 200), dtype=tf.uint8)
9 plt.imshow(img_data2.eval())
10 plt.show()
tf.image.resize_image_with_crop_or_pad函数可以用来进行图像裁剪或扩展，这个是由用户的目标宽度和高度决定的，另外无论是裁剪还是扩展都是从图片中心为基准的。
指定位置裁剪
1 import matplotlib.pyplot as plt
2 import tensorflow as tf
3
4 raw_data = tf.gfile.FastGFile(‘./new.jpg‘,‘rb‘).read()
5
6 with tf.Session() as sess:
7 img_data = tf.image.decode_jpeg(raw_data)
8 img_data2 = tf.cast(tf.image.crop_to_bounding_box(img_data, 0, 0, 200, 200), dtype=tf.uint8)
9 plt.imshow(img_data2.eval())
10 plt.show()
11
12 ~
图像上画框
1 import matplotlib.pyplot as plt
2 import tensorflow as tf
3
4 raw_data = tf.gfile.FastGFile(‘./new.jpg‘,‘rb‘).read()
5
6 with tf.Session() as sess:
7 img_data = tf.cast(tf.expand_dims(tf.image.decode_jpeg(raw_data), 0), tf.float32)
8 boxes = tf.constant([[[0.4, 0.4, 0.5, 0.5], [0.5, 0.5, 0.6, 0.6]]])
9 img_data2 = tf.cast(tf.image.draw_bounding_boxes(img_data, boxes), dtype=tf.uint8)
10 plt.imshow(img_data2.eval()[0])
11 plt.show()
