掩码到边界框的转换

292 阅读1分钟

本篇是对 Airbus Ship Detection Challenge的kernels中 From masks to bounding boxes 

的一个总结。

在将掩码转换乘图像的格式之后,导入skimage.measure,对掩码区域进行标签(1,2,3.......)(原理是对连通域进行分类标签),然后对标签的区域进行regionprops,得到目标狂的左上角和右下角的坐标值。利用cv2.rectangle对图像中的目标进行边界框的标注。

最后再对边界框进行文件的储存。