python3 opencv 圖象灰度化處理

圖象灰度化處理

src

# -*- coding:utf-8 -*-
# /usr/bin/python
'''
Author:Yan Errol  Email:[email protected]   Wechat:qq260187357
Date:2019-05-08--10:07
File：img2gray.py
Describe:將加載的圖象進行灰度化處理
'''

print (__doc__)

import cv2 as cv
import numpy as np
from matplotlib import pyplot as plt

def read_img(path):
    # Load an color image in grayscale
    img = cv.imread(path, 0)
    return img

def show_img(img):
    # matplot plt show img
    plt.imshow(img, cmap='gray', interpolation='bicubic')
    plt.xticks([]), plt.yticks([])  # to hide tick values on X and Y axis
    plt.show()

def show_Digit(vector):
    # 重新變形
    img = vector.reshape((8, 8))
    plt.imshow(img, cmap='gray')
    plt.show()



def img2vect(img):
    # 將灰度圖變爲向量
    # 變換爲 8×8
    img = cv.resize(img, (8, 8), interpolation=cv.INTER_LINEAR)
    # 變爲向量 並將數值放縮在 0-16之間
    return np.reshape(img, (64)) / 16

def write_img(img):
    # save img
    cv.imwrite('../data/test/test13.bmp', img)

def main():
    path = "../data/test/test15.bmp"
    img = read_img(path)
    vector = img2vect(img)
    print (vector)
    show_img(img)
    write_img(img)


if __name__=="__main__":
    main()

效果