python 分割 画图像的轮廓

<span style="font-family:Comic Sans MS;font-size:18px;">python 分割 画图像的轮廓</span>

</pre><pre code_snippet_id="1881684" snippet_file_name="blog_20160915_3_5459447" name="code" class="python">import numpy as np
import cv2

__author__ = 'xinyang'

Img_N = 130
algo_seg_folder = 'G:/ctx_2'
gt_seg_folder = 'G:/gt'
src_us_folder = 'G:/src_img'
save_folder = 'G:/only_gt_fuse'

for k in range(Img_N):
# for k in range(2):
print 'No. %d image...' % k
src_us_name = src_us_folder + '/' + str(k) + '.png'
gt_seg_name = gt_seg_folder + '/' + str(k) + '.png'
algo_seg_name = algo_seg_folder + '/' + str(k) + '.png'

src_us = cv2.imread(src_us_name, 1)
gt_seg = cv2.imread(gt_seg_name, 0)
algo_seg = cv2.imread(algo_seg_name, 0)

# get contour
gt_ret, gt_binImg = cv2.threshold(gt_seg, 127, 255, 0)
gt_im2, gt_contours, gt_hierarchy = cv2.findContours(gt_binImg, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)
algo_ret, algo_binImg = cv2.threshold(algo_seg, 127, 255, 0)
algo_im2, algo_contours, algo_hierarchy = cv2.findContours(algo_binImg, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)

# draw contour
cv2.drawContours(src_us, gt_contours, -1, (0, 0, 255), 2)
# cv2.drawContours(src_us, algo_contours, -1, (0, 255, 0), 2)

# save
save_name = save_folder + '/' + str(k) + '.png'
cv2.imwrite(save_name, src_us)