使用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)

总结

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

您可能感兴趣的文章:

python如何复制别人的虚拟环境
这篇文章主要介绍了python如何复制别人的虚拟环境问题,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教
2024-05-05
Python实现身份证前六位地区码对照表文件
这篇文章主要为大家详细介绍了如何利用Python实现身份证前六位地区码对照表文件，文中的示例代码讲解详细，感兴趣的可以了解一下
2022-12-12
基于python操作ES实例详解
这篇文章主要介绍了基于python操作ES实例详解,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
2019-11-11
详解Python中import机制
这篇文章主要介绍了Python中import机制的相关资料，帮助大家更好的理解和学习python，感兴趣的朋友可以了解下
2020-09-09
Python中字符串操作技巧
在编程中,字符串操作是一项基本技能,常见的操作包括大小写转换、字符串替换、倒序、统计、切片、切割、清理和对齐等,查找方法如find()、rfind()、index()和rindex()用于定位子串,字符串可以通过"+"号或join()方法拼接,去重技巧等,都是常见的字符串处理需求
2024-10-10
详解Python字符串原理与使用的深度总结
本文将学习字符串数据类型相关知识，将讨论如何声明字符串数据类型，字符串数据类型与 ASCII 表的关系，字符串数据类型的属性，以及一些重要的字符串方法和操作，超级干货，不容错过
2022-05-05
python 实时得到cpu和内存的使用情况方法
今天小编就为大家分享一篇python 实时得到cpu和内存的使用情况方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2018-06-06
python自动化测试通过日志3分钟定位bug
软件开发中通过日志记录程序的运行情况是一个开发的好习惯，对于错误排查和系统运维都有很大帮助，Python标准库自带了强大的logging日志模块，在各种python模块中得到广泛应用
2021-11-11
django框架模板语言使用方法详解
这篇文章主要介绍了django框架模板语言使用方法,结合实例形式详细分析了Django框架常用的模板语言语法与相关使用技巧,需要的朋友可以参考下
2019-07-07
Django项目单字段区间查询的实现
在Django项目中会碰到一些需求就是查询某个表中的一些字段从某日到某日的数据,你可以像在SQL中那样使用SELECT语句来查找指定字段,本文就来介绍两种方法,感兴趣的可以了解一下
2023-10-10