search

z-x-yang / Segment-and-Track-Anything

An open-source project dedicated to tracking and segmenting any objects in videos, either automatically or interactively. The primary algorithms utilized include the Segment Anything Model (SAM) for key-frame segmentation and Associating Objects with Transformers (AOT) for efficient tracking and propagation purposes.
GNU Affero General Public License v3.0
2.77k stars 334 forks source link

更换分割模型 #96

Closed xyj176 closed 10 months ago

xyj176 commented 1 year ago

能不能把SAM换成我自己训练的语义分割模型，该怎么替换

renyiyu commented 12 months ago

可以的，你把用sam预测的mask换成你自己语义分割模型的mask就行，然后按obj的类别叠在同一张mask（h,w)的格式上就行。