[英]Trimming the white space in an image using PIL in python
我正在使用SciKit-learn进行手写数字识别,因此我需要裁剪单击的图片,因此我在Word上准备了模板。 现在我想要的是沿着边框裁剪图像,以便可以进一步裁剪以提取数字。
示例图片如下:
为了裁剪图像,我正在使用此代码。
注意:父图像也有边框(在图像中不可见),因此修剪空白可能有助于获取修改后的父图像,以便对要进行的各种作物的预定义(高度,宽度)几乎相同图片。
您可以应用此管道:转换为灰度->应用阈值处理(转换为白色和黑色)->查找轮廓->选择正确形状的轮廓。
这是示例代码:
#!/usr/bin/env python
import cv2
BLACK_THRESHOLD = 200
THIN_THRESHOLD = 10
ANNOTATION_COLOUR = (222,0,222)
img = cv2.imread('template.png')
orig = img.copy()
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
thresh = cv2.threshold(gray, thresh=BLACK_THRESHOLD, maxval=255, type=cv2.THRESH_BINARY_INV)[1]
# Optional: save thesholded image
cv2.imwrite("temp_thres.png", thresh)
# Find contours on the thresholded image
contours = cv2.findContours(thresh,cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)[1]
for cont in contours:
# Find bounding rectangle of a contour
x,y,w,h = cv2.boundingRect(cont)
# Skip thin contours (vertical and horizontal lines)
if h<THIN_THRESHOLD or w<THIN_THRESHOLD:
continue
# Does the countour has the right shape (roughly four times longer than high)?
if 3*h<w<5*h:
roi = orig[y:y+h,x:x+w]
cv2.imwrite("four_letters.png",roi)
# Optional: draw annotations
cv2.rectangle(img,(x,y),(x+w,y+h),ANNOTATION_COLOUR,3)
# Optional: save annotated image
cv2.imwrite("temp_cont.png",img)
(您可以删除这三个可选步骤。它们只是用于生成图像temp_thres.png
和temp_cont.png
。)
输入图像template.png
:
阈值图像temp_thres.png
:
找到轮廓temp_cont.png
:
四个字母的空间four_letters.png
:
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.