分割一切，跟踪一切 Segment-and-Track Anything 使用笔记

AI算法网奇

已于 2023-11-13 21:29:30 修改

阅读量1.8k

点赞数

分类专栏：深度学习宝典文章标签：计算机视觉

于 2023-04-22 01:30:01 首次发布

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

版权

深度学习宝典专栏收录该内容

987 篇文章 713 订阅 ¥29.90 ¥99.00

订阅专栏

超级会员免费看

SAM-Track是浙江大学ReLER实验室开源的项目，它扩展了Meta AI的SAM模型，实现了视频目标分割和跟踪。该项目支持自动化和交互式分割跟踪，能处理多种场景，包括街景、AR、细胞、动画、航拍等，并提供WebUI方便使用。即将发布的V1版本将增加交互式视频分割功能。

摘要由CSDN通过智能技术生成

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

了解本专栏