zhang-tao-whu / e2ec

E2EC: An End-to-End Contour-based Method for High-Quality High-Speed Instance Segmentation
Other
214 stars 45 forks source link

检测框可视化 #46

Closed xingxg613 closed 2 months ago

xingxg613 commented 2 months ago

1 请问为什么检测框和分割的轮廓差别如此大? 检测框可视化的代码使用的是snake仓库中的代码, box = output['detection'][:, :4].detach().cpu().numpy() for i in range(len(ex)): color = next(colors).tolist() poly = ex[i] poly = np.append(poly, [poly[0]], axis=0) ax.plot(poly[:, 0], poly[:, 1], color=color, linewidth=5)

        x_min, y_min, x_max, y_max = box[i]
        ax.plot([x_min, x_min, x_max, x_max, x_min], [y_min, y_max, y_max, y_min, y_min], color='w', linewidth=0.5)

希望可以收到回复,非常感谢