163 lines
7.0 KiB
Python
163 lines
7.0 KiB
Python
import os
|
|
import sys
|
|
import cv2
|
|
import numpy as np
|
|
from matplotlib import pyplot as plt
|
|
|
|
|
|
def menu():
|
|
print("Выберите пункт меню:\n1.Linear filter\n2.Blur\n3.Median blur\n4.GaussianBlur\n5.Erode")
|
|
print("6.Dilate\n7.MORPH_OPERATIONS\n8.Sobel\n9.Laplacian\n10.Canny\n11.CalcHist\n12.EqualizeHist")
|
|
print("13.Save\n14.Add Figure\n15.Del Figure\n16.Exit")
|
|
|
|
|
|
def main():
|
|
image1 = cv2.imread("D:\\MACH\\LAB_2\\image0.jpg")
|
|
image2 = image1.copy()
|
|
im = image1.copy()
|
|
while True:
|
|
menu()
|
|
try:
|
|
s = int(input())
|
|
if s == 1:
|
|
image2 = image1.copy()
|
|
kernel = np.array([[-0.1, 0.2, -0.1], [0.2, 3.0, 0.2], [-0.1, 0.2, -0.1]])
|
|
image2 = cv2.filter2D(image2, -1, kernel)
|
|
cv2.imshow("Image_load", image1)
|
|
cv2.imshow("Result_Image", image2)
|
|
cv2.waitKey(0)
|
|
if s == 2:
|
|
image2 = image1.copy()
|
|
image2 = cv2.blur(image2, (5, 5))
|
|
cv2.imshow("Image_load", image1)
|
|
cv2.imshow("Result_Image", image2)
|
|
cv2.waitKey(0)
|
|
if s == 3:
|
|
image2 = image1.copy()
|
|
image2 = cv2.medianBlur(image2, 5)
|
|
cv2.imshow("Image_load", image1)
|
|
cv2.imshow("Result_Image", image2)
|
|
cv2.waitKey(0)
|
|
if s == 4:
|
|
image2 = image1.copy()
|
|
image2 = cv2.GaussianBlur(image2, (9, 9), cv2.BORDER_DEFAULT)
|
|
cv2.imshow("Image_load", image1)
|
|
cv2.imshow("Result_Image", image2)
|
|
cv2.waitKey(0)
|
|
if s == 5:
|
|
image2 = image1.copy()
|
|
image2 = cv2.erode(image2, np.ones((11, 11)))
|
|
cv2.imshow("Image_load", image1)
|
|
cv2.imshow("Result_Image", image2)
|
|
cv2.waitKey(0)
|
|
if s == 6:
|
|
image2 = image1.copy()
|
|
image2 = cv2.dilate(image2, np.ones((11, 11)))
|
|
cv2.imshow("Image_load", image1)
|
|
cv2.imshow("Result_Image", image2)
|
|
cv2.waitKey(0)
|
|
if s == 7:
|
|
image2 = image1.copy()
|
|
image3 = cv2.cvtColor(image2, cv2.COLOR_BGR2GRAY)
|
|
cv2.imshow("Image_load", image1)
|
|
kernel = np.ones((6, 6), np.uint8)
|
|
image3 = cv2.morphologyEx(image3, cv2.MORPH_OPEN, kernel, iterations=1)
|
|
cv2.imshow("MORPH_OPEN", image3)
|
|
image3 = cv2.cvtColor(image2, cv2.COLOR_BGR2GRAY)
|
|
image3 = cv2.morphologyEx(image3, cv2.MORPH_CLOSE, kernel, iterations=1)
|
|
cv2.imshow("MORPH_CLOSE", image3)
|
|
image3 = cv2.cvtColor(image2, cv2.COLOR_BGR2GRAY)
|
|
image3 = cv2.morphologyEx(image3, cv2.MORPH_GRADIENT, kernel, iterations=1)
|
|
cv2.imshow("MORPH_GRADIENT", image3)
|
|
image3 = cv2.cvtColor(image2, cv2.COLOR_BGR2GRAY)
|
|
image3 = cv2.morphologyEx(image3, cv2.MORPH_TOPHAT, kernel, iterations=1)
|
|
cv2.imshow("MORPH_TOPHAT", image3)
|
|
image3 = cv2.cvtColor(image2, cv2.COLOR_BGR2GRAY)
|
|
image3 = cv2.morphologyEx(image3, cv2.MORPH_BLACKHAT, kernel, iterations=1)
|
|
cv2.imshow("MORPH_BLACKHAT", image3)
|
|
image3 = cv2.cvtColor(image2, cv2.COLOR_BGR2GRAY)
|
|
cv2.waitKey(0)
|
|
if s == 8:
|
|
image2 = image1.copy()
|
|
image3 = cv2.cvtColor(image2, cv2.COLOR_BGR2GRAY)
|
|
image3 = cv2.GaussianBlur(image3, (3, 3), 0)
|
|
im3 = cv2.Sobel(image3, cv2.CV_64F, 1, 0, ksize=5)
|
|
im4 = cv2.Sobel(image3, cv2.CV_64F, 0, 1, ksize=5)
|
|
im5 = cv2.Sobel(image3, cv2.CV_8UC1, 0, 1, ksize=5)
|
|
cv2.imshow("Image_load", image1)
|
|
cv2.imshow("Result_Image_X", im3) # x
|
|
cv2.imshow("Result_Image_Y", im4) # y
|
|
cv2.imshow("Result_Gradient", im5) # gradient
|
|
cv2.waitKey(0)
|
|
if s == 9:
|
|
image2 = image1.copy()
|
|
image3 = cv2.cvtColor(image2, cv2.COLOR_RGB2GRAY)
|
|
image3 = cv2.GaussianBlur(image3, (3, 3), 0)
|
|
im3 = cv2.Laplacian(image3, cv2.CV_64F)
|
|
cv2.imshow("Image_load", image1)
|
|
cv2.imshow("Result_Image", im3) # x
|
|
cv2.waitKey(0)
|
|
if s == 10:
|
|
image2 = image1.copy()
|
|
image3 = cv2.cvtColor(image2, cv2.COLOR_RGB2GRAY)
|
|
image3 = cv2.blur(image2, (5, 5))
|
|
image3 = cv2.Canny(image3, 100, 100)
|
|
cv2.imshow("Image_load", image1)
|
|
cv2.imshow("Result_Image", image3) # x
|
|
cv2.waitKey(0)
|
|
if s == 11:
|
|
image2 = image1.copy()
|
|
color = ('b', 'g', 'r')
|
|
cv2.imshow("Image_load", image1)
|
|
for i, col in enumerate(color):
|
|
histr = cv2.calcHist([image2], [i], None, [256], [0, 256])
|
|
plt.plot(histr, color=col)
|
|
plt.xlim([0, 256])
|
|
plt.show()
|
|
plt.close()
|
|
cv2.waitKey(0)
|
|
if s == 12:
|
|
image2 = image1.copy()
|
|
image2 = cv2.cvtColor(image2, cv2.COLOR_RGB2GRAY)
|
|
image2 = cv2.equalizeHist(image2)
|
|
cv2.imshow("Image_load", image1)
|
|
cv2.imshow("Result_Image", image2)
|
|
cv2.waitKey(0)
|
|
if s == 13:
|
|
cv2.imshow("Image_load", image1)
|
|
cv2.imshow("Result_Image", image2)
|
|
cv2.imwrite("D:\\im1.jpg", image1)
|
|
cv2.imwrite("D:\\im2.jpg", image2)
|
|
cv2.waitKey(0)
|
|
if s == 14:
|
|
image2 = image1.copy()
|
|
cv2.imshow("Image_load", image1)
|
|
image2 = cv2.line(image2, (15, 15), (270, 270), (76, 187, 23), 10) # линия
|
|
image2 = cv2.rectangle(image2, (20, 20), (100, 100), (0, 0, 255), 10) # нарисуем четырехугольник
|
|
image2 = cv2.circle(image2, (100, 100), 40, (0, 255, 0), 10) # нарисуем круг
|
|
cv2.imshow("Result_Image", image2)
|
|
image1 = image2
|
|
cv2.waitKey(0)
|
|
if s == 15:
|
|
cv2.imwrite("1.jpg", im)
|
|
cv2.imwrite("2.jpg", image1)
|
|
img = cv2.imread("1.jpg")
|
|
mask = cv2.imread("2.jpg", 0)
|
|
res = cv2.bitwise_and(img, img, mask=mask)
|
|
image2 = res.copy()
|
|
image1 = res.copy()
|
|
cv2.imshow("Result_Image", res)
|
|
|
|
cv2.waitKey(0)
|
|
os.remove("1.jpg")
|
|
os.remove("2.jpg")
|
|
if s == 16:
|
|
cv2.destroyAllWindows()
|
|
sys.exit(0)
|
|
except ValueError:
|
|
print("Неверный пункт меню!!!! Выберите другое!")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|