分割api

分割api

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

import numpy as np
import cv2
import torch
import gc
import sys
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

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("Unknown model size in path!")


class SAM2_Track():

    def __init__(self,debug_dir,args):
        self.debug_dir=debug_dir

        model_cfg = dete