在图像上画框

最新推荐文章于 2023-09-05 17:10:23 发布

npupengsir

最新推荐文章于 2023-09-05 17:10:23 发布

阅读量224

点赞数

分类专栏： pytorch 画图

本文链接：https://blog.csdn.net/u012897374/article/details/117865355

版权

pytorch 同时被 2 个专栏收录

13 篇文章 0 订阅

订阅专栏

画图

3 篇文章 1 订阅

订阅专栏

def draw_mask(img, x1, x2, y1, y2, thickness):
	max_value = np.iinfo(img.dtype).max
    img[max(0, x1-thickness//2): min(img.shape[0], x1+thickness//2), max(0, y1-thickness//2): min(img.shape[1], y2+thickness//2)] = max_value 
    img[max(0, x2-thickness//2): min(img.shape[0], x2+thickness//2), max(0, y1-thickness//2): min(img.shape[1], y2+thickness//2)] = max_value 
    img[max(0, x1-thickness//2): min(img.shape[0], x2+thickness//2), max(0, y1-thickness//2): min(img.shape[1], y1+thickness//2)] = max_value 
    img[max(0, x1-thickness//2): min(img.shape[0], x2+thickness//2), max(0, y2-thickness//2): min(img.shape[1], y2+thickness//2)] = max_value 
    return img

img = draw_mask(img, x_min, x_max, y_min, y_max, thickness=10)