使用tensorflow将图片灰度化以及遇到的坑解决

更新时间：2024年02月27日 15:02:48 作者：hello689

这篇文章主要介绍了使用tensorflow将图片灰度化以及遇到的坑解决,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教

tensorflow将图片灰度化及遇到的坑

用tensorflow将图片灰度化，很简单的一个问题，几句代码就OK了。

但是这里边有很多坑，稍不留神，半天都不一定找得到原因。

我下面说说我遇到的坑

先给代码，再介绍我遇到的坑：

import tensorflow as tf
import numpy as np
import matplotlib.pyplot as plt
from keras_preprocessing.image import load_img,img_to_array
%matplotlib inline
 
 
image_name = 'cat1.jpg'
image = load_img(image_name)
#将图片转换为数组
image = img_to_array(image)
#转为黑白
image = tf.image.rgb_to_grayscale(image)
image = tf.squeeze(image,2)
plt.imshow(image)

步骤很简单：

读取彩色图片
转换为黑白图片
显示这个黑白图片

就这简简单单的几步，我却遇到了很多问题，不过在解决问题的过程中又学到了很多，还是很有收获的。

坑一

#坑一、不将image转换为数组，直接用rgb_to_grayscale方法，将图片转为灰度图片
image_name = 'cat1.jpg'
image = load_img(image_name)
#转为黑白
image = tf.image.rgb_to_grayscale(image)

执行上边代码会报错如下图，不能直接将一个图片转化为一个tensor。

这里需要先将image对象转化为数组array。

然后通过百度学会了img_to_array()方法。

图1、坑一错误信息

坑二

#坑二：最后图片无法显示
image_name = 'cat1.jpg'
image = load_img(image_name)
#将图片转换为数组
image = img_to_array(image)
#转为黑白
image = tf.image.rgb_to_grayscale(image)
plt.imshow(image)

错误信息：

图2、坑二错误

原图是三维的，图片维度为(480,500,3)，转为黑白后图片的shape变成了(480,500,1)。

猜测，黑白的是不是就两维。于是进行了降维操作学习了tf.squeeze()方法，最后降维后，大功告成，结束！

补充

我当时看了image的类型，本以为是因为转换为array后，里边值的类型为float，所以才不会显示图片。

当时就把float类型转为了int型。后来才发现，其实不管array里的值类型是float还是int,都会显示。

所以第一版成功运行的代码是把图片类型也转换了，代码如下，第六行多了一句将图片转为int型的代码。

image_name = 'cat1.jpg'
image = load_img(image_name)
plt.imshow(image)
#将图片转换为数组
image = img_to_array(image)
image = image.astype(dtype='uint8')
#转为黑白
image = tf.image.rgb_to_grayscale(image)
#降维
image = tf.squeeze(image,2)
plt.imshow(image)

总结

以上为个人经验，希望能给大家一个参考，也希望大家多多支持脚本之家。

您可能感兴趣的文章:

pyecharts如何旋转折线图的X轴标签
这篇文章主要介绍了pyecharts如何旋转折线图的X轴标签，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教
2022-11-11
Python中的np.vstack()和np.hstack()详解
这篇文章主要介绍了np.vstack()和np.hstack(),本文给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下
2023-04-04
Python googletrans库使用示例详解
googletrans是一个基于谷歌翻译API的Python库,支持多种语言的自动检测和翻译,提供了translate和detect方法,用于翻译文本和检测文本语言,通过简单的命令即可安装使用,适合需要实现多语言翻译功能的开发者
2024-09-09
python3实现ftp服务功能（服务端 For Linux）
这篇文章主要介绍了python3实现ftp服务功能，服务端 For Linux，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
2017-03-03
使用Python实现为PDF文件添加图章
在日常工作中,我们经常需要给PDF文档添加一些标识,比如公司的图章或水印图章,所以本文就来为大家详细介绍一下如何使用Python实现为PDF文件添加图章,需要的可以参考下
2023-11-11
Python找出9个连续的空闲端口
这篇文章主要介绍了Python找出9个连续的空闲端口的方法，感兴趣的小伙伴们可以参考一下
2016-02-02
python类函数的有效调用方式
文章介绍了Python中类的几种不同方法,并展示了如何有效调用这些方法,实验涵盖了属性方法、静态方法、类方法、普通方法、保护方法和私有方法,并详细说明了每种方法的调用方式,文章还讨论了如何处理调用错误和返回值,并提供了一些实用建议
2024-11-11
python GUI库图形界面开发之PyQt5打开保存对话框QFileDialog详细使用方法与实例
这篇文章主要介绍了python GUI库图形界面开发之PyQt5打开保存对话框QFileDialog详细使用方法与实例,需要的朋友可以参考下
2020-02-02
使用Python获取Linux系统的各种信息
这篇文章主要介绍了使用Python获取Linux系统的各种信息,例如系统类型、CPU信息、内存信息、块设备等,需要的朋友可以参考下
2014-07-07
python-parser.parse_args()解析参数问题
这篇文章主要介绍了python-parser.parse_args()解析参数问题,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教
2023-08-08