grounded_sam2 使用踩坑笔记

目录

自己封装的批量检测图片

封装的视频版推理代码：

---

检测可以设置阈值，提示词也可以设置阈值

检测框取了最大值：

自己封装的批量检测图片

mask做了压缩编码，保存为json

grounded_sam2_dao_img.py

import glob
import os
import cv2
import json
import torch
import numpy as np
import supervision as sv
import pycocotools.mask as mask_util
from pathlib import Path
import transformers
transformers.utils.logging.set_verbosity_error()  # 减少日志干扰
from torchvision.ops import box_convert
from sam2.build_sam import build_sam2
from sam2.sam2_image_predictor import SAM2ImagePredictor
from grounding_dino.groundingdino.util.inference impor