samurai roi

AI算法网奇

于 2025-06-30 00:58:10 发布

阅读量19

点赞数

CC 4.0 BY-SA版权

分类专栏： python宝典文章标签： opencv 人工智能计算机视觉

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

python宝典专栏收录该内容

1154 篇文章 ¥29.90 ¥99.00

订阅专栏

超级会员免费看

import argparse
import os
import os.path as osp
import time
from glob import glob

import numpy as np
import cv2
import torch
import gc
import torchvision
import sys

from scripts.roi_class import ROIDrawer

sys.path.append("./sam2")
from sam2.build_sam import build_sam2_video_predictor

color = [(255,255,255)]

from mask_dictionary_model import MaskDictionaryModel, ObjectInfo, TrackingResult
from ultralytics import YOLO
import  imageio
import pickle
def determine_model_cfg(model_path):
    if "large" in model_path:
        return "configs/samurai/sam2.1_hiera_l.yaml"
    elif "base_plus" in model_path:
        return "configs/samurai/sam2.1_hiera_b+.yaml"
    else:
        raise ValueError("Unk