Segment-and-Track Anything!视频版SAM来了,分割/跟踪/编辑一切,已开源!
目录
依赖项安装:
pip install spatial-correlation-sampler
跑通测试代码:
import os
import cv2
from SegTracker import SegTracker
from model_args import aot_args, sam_args, segtracker_args
from PIL import Image
from aot_tracker import _palette
import numpy as np
import torch
import imageio
import matplotlib.pyplot as plt
from scipy.ndimage import binary_dilation
import gc
def save_prediction(pred_mask, output_dir, file_name):
save_mask = Image.fromarray(pred_mask.astype(np