Tensorflow读取图片并转换格式_M_小蜗牛

个人资料

微博

正文字体大小：大中小

Tensorflow读取图片并转换格式

(2022-04-04 20:15:07)

分类： AITensorflow研究

import tensorflow as tf

import matplotlib.pylab as plt

print(tf.test.is_built_with_gpu_support())

print(tf.test.is_built_with_rocm())

data_path='./cat.jpg'

image_raw_data1 = tf.io.read_file(data_path,None)

image = tf.image.decode_image(contents=image_raw_data1,channels=3)

image_jpg = tf.image.decode_jpeg(image_raw_data1,channels=3)

#print(type(image))

#plt.imshow(image)

plt.imshow(image_jpg)

image_jpg = image_jpg.numpy()

print(type(image_jpg) )

print(image_jpg.shape)

(717, 719, 3)

阅读┊ 收藏 ┊ 喜欢 ▼ ┊打印┊举报/Report