Chinaunix首页 | 论坛 | 博客
  • 博客访问: 4510951
  • 博文数量: 356
  • 博客积分: 10458
  • 博客等级: 上将
  • 技术积分: 4734
  • 用 户 组: 普通用户
  • 注册时间: 2008-03-24 14:59
文章分类

全部博文(356)

文章存档

2020年(17)

2019年(9)

2018年(26)

2017年(5)

2016年(11)

2015年(20)

2014年(2)

2013年(17)

2012年(15)

2011年(4)

2010年(7)

2009年(14)

2008年(209)

分类: Python/Ruby

2020-05-30 13:18:53

对于彩色图像,不管其图像格式是PNG,还是BMP,或者JPG,在PIL中,使用Image模块的open()函数打开后,返回的图像对象的模式都是“RGB”。而对于灰度图像,不管其图像格式是PNG,还是BMP,或者JPG,打开后,其模式为“L”。
img = Image.open('1.bmp').convert('L')
如果convert中 模式“1”为二值图像,非黑即白。但是它每个像素用8个bit表示,0表示黑,255表示白。这个跟mnist很像,但是也有不同的
mnist 里的颜色是0代表白色(背景),1.0代表黑色

再转成28x28的大小
# resize的过程
if img.size[0] != 28 or img.size[1] != 28:
    img = img.resize((28, 28))

# 暂存像素值的一维数组
arr = []

for i in range(28):
    for j in range(28):
        # mnist 里的颜色是0代表白色(背景),1.0代表黑色, 这里把灰度图转换成mnist的数据
        pixel = 1.0 - float(img.getpixel((j, i)))/255.0
        # pixel = 255.0 - float(img.getpixel((j, i))) # 如果是0-255的颜色值
        arr.append(pixel)

#reshape转成四维的
arr1 = np.array(arr).reshape((1, 28, 28, 1))
print (model.predict(arr1))
阅读(2121) | 评论(0) | 转发(0) |
给主人留下些什么吧!~~