利用easygui实现一个简易Opencv的小系统

荆小南

import numpy as np
from matplotlib import pyplot as plt
import torch
import cv2 as cv
import easygui as g
import sys

def winname(name):
    return name.encode('gbk').decode(errors='ignore')
def fake_color_image(image_gray):
    r = 0
    g=0
    b=0
    height = image_gray.shape[0]
    weight = image_gray.shape[1]
    out_color_image = np.zeros((height,weight,3),np.uint8)
    for i in range(height):
        for j in range(weight):
             val = image_gray[i,j]
             if (val <128):
                 r = 0
             elif (val <192):
                 r=255/64*(val-128)
             else:
                 r = 255
             if (val < 64):
                 g = 255 / 64 * val
             elif (val < 192):
                 g = 255
             else:
                 g = -255 / 63 * (val - 192) + 255
             if (val < 64):
                 b = 255
             elif (val < 128):
                 b = -255 / 63 * (val - 192) + 255
             else:
                 b = 0
             out_color_image[i,j,0] = b
             out_color_image[i,j,1] = g
             out_color_image[i,j,2] = r
    cv.imshow("fake color image",out_color_image)
def image_gray1(img_gray):
    img_gray1 = img_gray.copy()
    img_gray1 = (img_gray[:, :, 0] + img_gray[:, :, 1] + img_gray[:, :, 2]) / 3  # 第一种方式为对各个通道求均值
    img_gray1 = img_gray1.astype(np.uint8)  # 猪油数组类型为unit类型时，才会认为是一张图片
    cv.imshow("gray image 1",img_gray1)
    cv.waitKey(1000)
def image_gray2(img_gray):
    img_gray2 = img_gray.copy()
    img_gray2 = img_gray[:, :, 0] * 0.11 + img_gray[:, :, 1] * 0.59 + img_gray[:, :, 2] * 0.3
    img_gray2 = img_gray2.astype(np.uint8)  # GRAY=0.3*R+0.59*G+0.11*B：
    cv.imshow("gray image 2",img_gray2)
    cv.waitKey(10)

plt.rcParams['font.sans-serif']=['SimHei']   # 用黑体显示中文
plt.rcParams['axes.unicode_minus']=False     # 正常显示负号
a=''
filename = ''
img =  np.ones((3,3),dtype=np.uint8)
msg="请输入您想要完成的任务（建议您第一步先打开图片）"
title='第一次作业'
choice=('打开图片','退出')
a=g.buttonbox(msg=msg,title=title,choices=choice)
if a == '打开图片':
        filename = g.fileopenbox(msg="请打开一个jpg文件")
        img = cv.imread(filename)
        msg1 = "选择您想要实现的功能"
        title1 = '第一次作业'
        choice1 = ('灰度化1', '灰度化2', '伪彩色', '直方图', '显示图片', '退出')
        q=1
        while q:
            b=g.buttonbox(msg=msg1,title=title1,choices=choice1)
            #while b!='退出':
            if b == '灰度化1':
                image_gray1(img)
            elif b == '灰度化2':
                image_gray2(img)
            elif b == '显示图片':
                cv.imshow("original image", img)
                cv.waitKey(1000)
            elif b == '伪彩色':
                img_gray = img[:, :, 0] * 0.11 + img[:, :, 1] * 0.59 + img[:, :, 2] * 0.3
                img_gray = img_gray.astype(np.uint8)              #GRAY=0.3*R+0.59*G+0.11*B：
                fake_color_image(img_gray)  #显示伪彩色图
            elif b == '直方图':
                img_gray = img[:, :, 0] * 0.11 + img[:, :, 1] * 0.59 + img[:, :, 2] * 0.3
                img_gray = img_gray.astype(np.uint8)
                pix = []
                height = img_gray.shape[0]
                weight = img_gray.shape[1]
                for i in range(height):
                    for j in range(weight):
                        pix.append(int(img_gray[i,j]))
                # pix1=[]
                # for i in range(256):
                #      number = pix.count(i)
                #      pix1.append(number)
                pix = np.array(pix)

                # data:必选参数，绘图数据
                # bins:直方图的长条形数目，可选项，默认为10
                # normed:是否将得到的直方图向量归一化，可选项，默认为0，代表不归一化，显示频数。normed=1，表示归一化，显示频率。
                # facecolor:长条形的颜色
                # edgecolor:长条形边框的颜色
                # alpha:透明度
                plt.hist(pix,bins=256,range=[0,255])     #绘制直方图（第二种方式的）

                plt.xlabel( " gray number " )
                plt.ylabel( " number " )
                plt.title("灰度直方图")
                plt.show()
            else:
                q=0
cv.waitKey(0)

weiter

放到代码框里咯？那样方便阅读

荆小南

1. import numpy as np
   
2. from matplotlib import pyplot as plt
   
3. import torch
   
4. import cv2 as cv
   
5. import easygui as g
   
6. import sys
   
7. 
   
8. def winname(name):
   
9.     return name.encode('gbk').decode(errors='ignore')
   
10. def fake_color_image(image_gray):
    
11.     r = 0
    
12.     g=0
    
13.     b=0
    
14.     height = image_gray.shape[0]
    
15.     weight = image_gray.shape[1]
    
16.     out_color_image = np.zeros((height,weight,3),np.uint8)
    
17.     for i in range(height):
    
18.         for j in range(weight):
    
19.              val = image_gray[i,j]
    
20.              if (val <128):
    
21.                  r = 0
    
22.              elif (val <192):
    
23.                  r=255/64*(val-128)
    
24.              else:
    
25.                  r = 255
    
26.              if (val < 64):
    
27.                  g = 255 / 64 * val
    
28.              elif (val < 192):
    
29.                  g = 255
    
30.              else:
    
31.                  g = -255 / 63 * (val - 192) + 255
    
32.              if (val < 64):
    
33.                  b = 255
    
34.              elif (val < 128):
    
35.                  b = -255 / 63 * (val - 192) + 255
    
36.              else:
    
37.                  b = 0
    
38.              out_color_image[i,j,0] = b
    
39.              out_color_image[i,j,1] = g
    
40.              out_color_image[i,j,2] = r
    
41.     cv.imshow("fake color image",out_color_image)
    
42. def image_gray1(img_gray):
    
43.     img_gray1 = img_gray.copy()
    
44.     width = img_gray1.shape[1]
    
45.     height = img_gray1.shape[0]
    
46.     for i in range(height):
    
47.         for j in range(width):
    
48.             img_gray1[i,j]= (int(img_gray[i,j, 0]) + int(img_gray[i,j, 1]) + int(img_gray[i,j, 2]))/ 3  #强行int类型，实现了灰度化
    
49.     # img_gray1 = (img_gray[:, :, 0] + img_gray[:, :, 1] + img_gray[:, :, 2]) / 3  # 第一种方式为对各个通道求均值
    
50.     img_gray1 = img_gray1.astype(np.uint8)  # 猪油数组类型为unit类型时，才会认为是一张图片
    
51.     cv.imshow("gray image 1",img_gray1)
    
52.     cv.waitKey(1000)
    
53. def image_gray2(img_gray):
    
54.     img_gray2 = img_gray.copy()
    
55.     img_gray2 = img_gray[:, :, 0] * 0.11 + img_gray[:, :, 1] * 0.59 + img_gray[:, :, 2] * 0.3
    
56.     img_gray2 = img_gray2.astype(np.uint8)  # GRAY=0.3*R+0.59*G+0.11*B：
    
57.     cv.imshow("gray image 2",img_gray2)
    
58.     cv.waitKey(10)
    
59. 
    
60. plt.rcParams['font.sans-serif']=['SimHei']   # 用黑体显示中文
    
61. plt.rcParams['axes.unicode_minus']=False     # 正常显示负号
    
62. a=''
    
63. filename = ''
    
64. img =  np.ones((3,3),dtype=np.uint8)
    
65. msg="请输入您想要完成的任务（建议您第一步先打开图片）"
    
66. title='第一次作业'
    
67. choice=('打开图片','退出')
    
68. a=g.buttonbox(msg=msg,title=title,choices=choice)
    
69. if a == '打开图片':
    
70.         filename = g.fileopenbox(msg="请打开一个jpg文件")
    
71.         img = cv.imread(filename)
    
72.         msg1 = "选择您想要实现的功能"
    
73.         title1 = '第一次作业'
    
74.         choice1 = ('灰度化1', '灰度化2', '伪彩色', '直方图', '显示图片', '退出')
    
75.         q=1
    
76.         while q:
    
77.             b=g.buttonbox(msg=msg1,title=title1,choices=choice1)
    
78.             #while b!='退出':
    
79.             if b == '灰度化1':
    
80.                 image_gray1(img)
    
81.             elif b == '灰度化2':
    
82.                 image_gray2(img)
    
83.             elif b == '显示图片':
    
84.                 cv.imshow("original image", img)
    
85.                 cv.waitKey(1000)
    
86.             elif b == '伪彩色':
    
87.                 img_gray = img[:, :, 0] * 0.11 + img[:, :, 1] * 0.59 + img[:, :, 2] * 0.3
    
88.                 img_gray = img_gray.astype(np.uint8)              #GRAY=0.3*R+0.59*G+0.11*B：
    
89.                 fake_color_image(img_gray)  #显示伪彩色图
    
90.             elif b == '直方图':
    
91.                 img_gray = img[:, :, 0] * 0.11 + img[:, :, 1] * 0.59 + img[:, :, 2] * 0.3
    
92.                 img_gray = img_gray.astype(np.uint8)
    
93.                 pix = []
    
94.                 height = img_gray.shape[0]
    
95.                 weight = img_gray.shape[1]
    
96.                 for i in range(height):
    
97.                     for j in range(weight):
    
98.                         pix.append(int(img_gray[i,j]))
    
99.                 pix1=[]
    
100.                 for i in range(256):
     
101.                      number = pix.count(i)
     
102.                      pix1.append(number)
     
103.                 # pix = np.array(pix)
     
104.                 pix1 = np.array(pix1)
     
105. 
     
106.                 # data:必选参数，绘图数据
     
107.                 # bins:直方图的长条形数目，可选项，默认为10
     
108.                 # normed:是否将得到的直方图向量归一化，可选项，默认为0，代表不归一化，显示频数。normed=1，表示归一化，显示频率。
     
109.                 # facecolor:长条形的颜色
     
110.                 # edgecolor:长条形边框的颜色
     
111.                 # alpha:透明度
     
112.                 # plt.hist(pix,bins=256,range=[0,255])     #绘制直方图（第二种方式的）
     
113.                 plt.plot(pix1)              #利用plot直接根据每个像素点的个数直接绘制直方图
     
114.                 plt.xlabel( " gray number " )
     
115.                 plt.ylabel( " number " )
     
116.                 plt.title("灰度直方图")
     
117.                 plt.show()
     
118.             else:
     
119.                 q=0
     
120. cv.waitKey(0)

复制代码

荆小南

是在抱歉，第一次投稿，有很多不清楚的地方还请见谅