01.图像分割中将掩码转换为坐标点的方法(单目标)

1.场景

图像分割中,模型输出二值掩码图,转化为目标检测中的检测框坐标

2.实现

该方法只适用于单目标

 import numpy as np
 def mask2box(self, mask):   # [x1,y1,x2,y2]
        '''从mask反算出其边框
        mask:[h,w]  0、1组成的图片
        1对应对象,只需计算1对应的行列号(左上角行列号,右下角行列号,就可以算出其边框)
        '''
        index = np.argwhere(mask == 1)
        rows = index[:, 0]
        clos = index[:, 1]
        # 解析左上角行列号
        left_top_r = np.min(rows)  # y
        left_top_c = np.min(clos)  # x
        # 解析右下角行列号
        right_bottom_r = np.max(rows) # y
        right_bottom_c = np.max(clos) # x
        return left_top_c,left_top_r,right_bottom_c,right_bottom_r

你可能感兴趣的:(目标检测,计算机视觉,深度学习,目标检测,opencv)