segment-anything 根据box分割

AI算法网奇

已于 2023-04-10 11:29:02 修改

阅读量1k

点赞数 1

分类专栏：深度学习宝典文章标签： python numpy 深度学习

于 2023-04-10 11:17:14 首次发布

本文链接：https://blog.csdn.net/jacke121/article/details/130055723

版权

深度学习宝典专栏收录该内容

987 篇文章 713 订阅 ¥29.90 ¥99.00

订阅专栏

超级会员免费看

本文介绍了如何利用Python和深度学习技术，通过指定的box进行交互式分割和批量预测。提供了针对单个box的分割代码以及适用于多个box的批量预测流程。

摘要由CSDN通过智能技术生成

整理了根据box分割box内容，如下图，可以用来分割标注：

一个box，交互分割代码：

多个box批量预测：

一个box，交互分割代码：

import time

using_colab = False
import numpy as np
import torch
# import matplotlib.pyplot as plt
import cv2

def show_points(coords, labels, ax, marker_size=375):
    pos_points = coords[labels == 1]
    neg_points = coords[labels == 0]
    ax.scatter(pos_points[:, 0], pos_points[:, 1], color='green', marker='*', s=marker_size, edg

了解本专栏