Question

我写了一个代码，找到公用事业仪表测量结果部分并将其裁剪。问题是它会调整图像大小。在不调整大小的情况下如何做，最后需要找到图像的已找到部分的参数，例如宽度，高度，中心x和y。

import numpy as np
import cv2
import sys
from PIL import Image
import imutils
import os.path 

if len(sys.argv) != 3:
    print ("%s input_file output_file" , sys.argv[0])
    sys.exit()
else:
    input_file = sys.argv[1]
    output_file = sys.argv[2]

if not os.path.isfile(input_file):
    print ("No such file '%s'" , input_file)
    sys.exit()

img = cv2.imread(input_file)
img = imutils.resize(img, height=500)

## Change to LAB space
lab = cv2.cvtColor(img, cv2.COLOR_BGR2LAB)
l,a,b = cv2.split(lab)
imglab = np.hstack((l,a,b))

## normalized the a channel to all dynamic range
na = cv2.normalize(a, None, alpha=0, beta=255, norm_type=cv2.NORM_MINMAX, dtype=cv2.CV_8UC1) 

## Threshold to binary
retval, threshed = cv2.threshold(na, thresh = 180,  maxval=255, type=cv2.THRESH_BINARY)    
## Do morphology
kernel = cv2.getStructuringElement( cv2.MORPH_ELLIPSE , (3,3))
opened = cv2.morphologyEx(threshed, cv2.MORPH_OPEN,kernel)
res = np.hstack((threshed, opened)) 
## Find contours
_, contours, hierarchy = cv2.findContours(opened, mode=cv2.RETR_LIST, method=cv2.CHAIN_APPROX_SIMPLE)    
## Draw Contours
res = img.copy()
cv2.drawContours(res, contours, -1, (255,0,0), 10)
## Filter Contours
for idx, contour in enumerate(contours):
    bbox = cv2.boundingRect(contour)
    area = bbox[-1]*bbox[-2]
    if area < 2150:
        continue
    (x, y, w, h) = cv2.boundingRect(contour)
    print(x, y, w, h)
    cv2.rectangle(res, (0, y), (x + w, y + h), (0, 255, 0), 4)
    crop_img = img[y:y+h*2, 0:x+w]
    (w, h, c) = img.shape
    print(w, h)
cv2.imwrite(output_file, crop_img)

输入图片为enter image description here

在公用事业仪表图像中使用Python中的OpenCV进行对象识别

0 个答案: