C.THRESHOLD_MIN_REC_EVENNESS, C.THRESHOLD_MAX_LINE_THICKNESS,
C.THRESHOLD_MIN_LIN_LENGTH, C.THRESHOLD_MAX_IMG_DENT_RATIO)
# get corner of boundaries -> img or block check
corners_rec = det.get_corner(boundary_rec)
corners_block, corners_img = det.block_or_img(
binary, corners_rec, C.THRESHOLD_MAX_BLOCK_BORDER_THICKNESS,
C.THRESHOLD_MAX_BLOCK_CROSS_POINT)
# refine img component
corners_img = det.img_refine2(corners_img, C.THRESHOLD_MAX_IMG_EDGE_RATIO,
C.THRESHOLD_MUST_IMG_HEIGHT,
C.THRESHOLD_MUST_IMG_WIDTH)
# *** Step 3 *** post-processing: remove img elements from original image and segment into smaller size
img_clean = draw.draw_bounding_box(corners_img, org, (255, 255, 255), -1)
seg.segment_img(img_clean, 600, out_img_segment, 0)
# draw results
draw_bounding = draw.draw_bounding_box(corners_block, org, (0, 255, 0))
draw_bounding = draw.draw_bounding_box(corners_img, draw_bounding,
(0, 0, 255))
# draw_boundary = draw.draw_boundary(boundary_rec, org.shape)
# save results
if is_save:
cv2.imwrite(out_img_draw, draw_bounding)
cv2.imwrite(out_img_gradient, binary)
cv2.imwrite(out_img_clean, img_clean)
file.save_corners(out_label, corners_block, 'div')
file.save_corners(out_label, corners_img, 'img', False)
end = file.timer(start)
print('Save ' + index + '\n')
corners_img = det.img_refine2(corners_img, C.THRESHOLD_MAX_EDGE_RATIO)
# remove img elements and segment into smaller size
img_clean = draw.draw_bounding_box(corners_img, org, (255, 255, 255), -1)
seg.segment_img(img_clean, 600, 'output/segment')
# draw results
draw_bounding = draw.draw_bounding_box(corners_block, org, (0, 255, 0))
draw_bounding = draw.draw_bounding_box(corners_img, draw_bounding, (0, 0, 255))
draw_boundary = draw.draw_boundary(boundary_rec, org.shape)
# save results
if is_save:
cv2.imwrite('output/org.png', org)
cv2.imwrite('output/labeled.png', draw_bounding)
cv2.imwrite('output/boundary.png', draw_boundary)
cv2.imwrite('output/gradient.png', binary)
cv2.imwrite('output/clean.png', img_clean)
file.save_corners('output/corners.csv', corners_block, 'div')
file.save_corners('output/corners.csv', corners_img, 'img', False)
# show results
if is_show:
cv2.imshow('org', draw_bounding)
cv2.imshow('boundary', draw_boundary)
cv2.imshow('gradient', binary)
cv2.imshow('clean', img_clean)
cv2.waitKey(0)
end = file.timer(start)