Spaces:
Running
Running
File size: 2,353 Bytes
a383d0e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 |
import cv2
import numpy as np
import lib_ip.block_division as blk
import lib_ip.ip_preprocessing as pre
import lib_ip.ip_detection as det
def nothing(x):
pass
def get_contour(org, binary):
def cvt_bbox(bbox):
'''
x,y,w,h -> colmin, rowmin, colmax, rowmax
'''
return bbox[0], bbox[1], bbox[0] + bbox[2], bbox[1] + bbox[3]
board = org.copy()
hie, contours, _ = cv2.findContours(binary, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
res_contour = []
for i in range(len(contours)):
if cv2.contourArea(contours[i]) < 200:
continue
cnt = cv2.approxPolyDP(contours[i], 0.001*cv2.arcLength(contours[i], True), True)
res_contour.append(cnt)
cv2.drawContours(board, res_contour, -1, (0,0,255), 1)
return board
img_file = 'E:\\Mulong\\Datasets\\rico\\combined\\1014.jpg'
resize_height = 800
cv2.namedWindow('control')
cv2.createTrackbar('resize_height', 'control', 800, 1600, nothing)
cv2.createTrackbar('grad_min', 'control', 4, 255, nothing)
cv2.createTrackbar('grad_min_blk', 'control', 5, 255, nothing)
cv2.createTrackbar('c1', 'control', 1, 1000, nothing)
cv2.createTrackbar('c2', 'control', 1, 1000, nothing)
while 1:
resize_height = cv2.getTrackbarPos('resize_height', 'control')
grad_min = cv2.getTrackbarPos('grad_min', 'control')
grad_min_blk = cv2.getTrackbarPos('grad_min_blk', 'control')
c1 = cv2.getTrackbarPos('c1', 'control')
c2 = cv2.getTrackbarPos('c2', 'control')
org, grey = pre.read_img(img_file, resize_height)
# org = cv2.medianBlur(org, 3)
# org = cv2.GaussianBlur(org, (3,3), 0)
binary = pre.binarization(org, grad_min)
binary_r = pre.reverse_binary(binary)
# blk.block_division(grey, grad_thresh=grad_min_blk, step_v=10, step_h=10, show=True)
cv2.imshow('bijn', binary)
cv2.imshow('r', binary_r)
cv2.waitKey(10)
# canny = cv2.Canny(grey, c1, c2)
# hie, contours, _ = cv2.findContours(binary, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
# b_contour = get_contour(org, binary)
# c_contour = get_contour(org, canny)
# b_contour = cv2.hconcat([b_contour, c_contour])
# binary = cv2.hconcat([binary, binary_r, canny])
# cv2.imshow('org', org)
# cv2.imshow('b_cnt', b_contour)
# cv2.imshow('bin', binary)
# cv2.imshow('canny', canny)
|