2017-09-13 75 views
0

使用下一個代碼,我可以拍攝一張圖像,並將其分成包含感興趣區域的各種較小圖像。從進程中排除某些尺寸的圖像(OpenCV,Python)

import cv2 
import sys 

sys.path.insert(0, 'C:\\Users\\Bob\\Desktop\\Project') 
sys.path.insert(0, 'C:\\Users\\Bob\\Desktop\\Project\\FOLDER') 
sys.path.insert(0, 'C:\\Users\\Bob\\Desktop\\Project\\READER') 

import FOLDER.folders 
import READER.extractor 

timestr = FOLDER.folders.timestr 

################## AREA 1 ################## 

# Load the image 
img = cv2.imread('C:\\Users\\Bob\\Desktop\\Destination\\' + str(timestr) + '\\EXTRACTED\\' + 'area1.png') 

# convert to grayscale 
gray = cv2.cvtColor(img,cv2.COLOR_BGR2GRAY) 

# smooth the image to avoid noises 
gray = cv2.medianBlur(gray,5) 

# Apply adaptive threshold 
thresh = cv2.adaptiveThreshold(gray,255,1,1,11,2) 
thresh_color = cv2.cvtColor(thresh,cv2.COLOR_GRAY2BGR) 

# apply some dilation and erosion to join the gaps - change iterations value to detect more or less area's 
thresh = cv2.dilate(thresh,None,iterations = 15) 
thresh = cv2.erode(thresh,None,iterations = 15) 

# Find the contours 
image,contours,hierarchy = cv2.findContours(thresh,cv2.RETR_TREE,cv2.CHAIN_APPROX_SIMPLE) 

# For each contour, find the bounding rectangle and draw it 
idx =0 

for cnt in contours: 
    idx += 1 
    x,y,w,h = cv2.boundingRect(cnt) 
    roi = gray[y:y + h, x:x + w] 
    cv2.rectangle(img,(x,y),(x+w,y+h),(0,255,0),2) 
    cv2.rectangle(thresh_color,(x,y),(x+w,y+h),(0,255,0),2) 
    cv2.imwrite('C:\\Users\\Bob\\Desktop\\Destination\\' + str(timestr) + '\\EXTRACTED\\ex_area1' + str(idx) + '.png',roi) 

這是一個例子:

圖像加載

origin

輸出

output1

output2

該代碼還給出了一些我不想要的小圖像(假設是工件)。所有這些圖像都低於一定的尺寸。

selection

我的問題是:我一定要添加到代碼上面有什麼要刪除這些圖片?要刪除下一個圖像,例如下一個尺寸:250(寬)X 60(高)像素?

謝謝

提示:使用此代碼來檢測領域:Improve text area detection (OpenCV, Python)

回答

1

只是不他們在第一時間寫。只寫60 * 250以上的圖像。

for cnt in contours: 
     idx += 1 
     x,y,w,h = cv2.boundingRect(cnt) 
     roi = gray[y:y + h, x:x + w] 
     cv2.rectangle(img,(x,y),(x+w,y+h),(0,255,0),2) 
     cv2.rectangle(thresh_color,(x,y),(x+w,y+h),(0,255,0),2) 
     if (w*h>250*60): #change this for proper settings 
      cv2.imwrite('C:\\Users\\Bob\\Desktop\\Destination\\' + str(timestr) 
      + '\\EXTRACTED\\ex_area1' + str(idx) + '.png',roi) 
+0

謝謝。它似乎工作。 – Link