sam2 samurai box分割图片封装

# coding=utf-8
import argparse
import gc
import glob
import os
import os.path as osp
import sys
import time
import traceback

import cv2
import numpy as np
import torch

current_dir = os.path.dirname(os.path.abspath(__file__))
current_dir_p = os.path.dirname(current_dir)
os.chdir(current_dir)
print('current_dir', current_dir)
paths = [current_dir, current_dir_p]

paths.extend(['./', "./sam2"])
for path in paths:
    sys.path.insert(0, path)
    os.environ['PYTHONPATH'] = (os.environ.get('PYTHONPATH', '') + ':' + path).strip(':')
import pickle
import platform

system_name = platform.system()
import imageio
import supervision as sv
from Config import Con