Kaggle Plot Ultrasoud Mask

04 July 2015

将kaggle上的超声波识别比赛的神经组织mask 合并到原始的超声波图片上。 原始的图片如下:

轮廓mask:

将上面两个图合并到一张图上去的代码如下:

import matplotlib.pyplot as plt
import cv2
import numpy as np  
img = plt.imread("1_1.tif")
mask = plt.imread("1_1_mask.tif")
img_color = np.dstack([img, img, img])
mask_pix = cv2.Canny(mask,200,100) > 0
img_color[mask_pix, 0] = 255
img_color[mask_pix, 1] = 255
img_color[mask_pix, 2] = 255
plt.imshow(img_color)
plt.show()

合并后的结果如下:

tips: tif文件灰度存为png灰度

import Image
Image.open('1_1.tif').convert('LA').save('1_1.png')

参考:kaggle script