Mirrors/Track-Anything
1
0
Fork 0
mirror of https://github.com/gaomingqi/Track-Anything.git synced 2025-12-16 08:27:49 +01:00
Code Issues Packages Projects Releases Wiki Activity

sam_model

Browse Source
This commit is contained in:
ShangGaoG
2023-04-17 19:57:47 +08:00
parent 1635a50e4f
commit 113299d99c
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
track_anything.py
View File

@@ -12,7 +12,7 @@ class TrackingAnything():
def __init__(self, sam_checkpoint, xmem_checkpoint, args): def __init__(self, sam_checkpoint, xmem_checkpoint, args):
self.args = args self.args = args
self.samcontroler = SamControler(sam_checkpoint, args.sam_model_type, args.device) self.samcontroler = SamControler(sam_checkpoint, args.sam_model_type, args.device)
self.xmem = BaseTracker(xmem_checkpoint, device=args.device, sam_checkpoint=sam_checkpoint, model_type=args.sam_model_type) self.xmem = BaseTracker(xmem_checkpoint, device=args.device, sam_model=sam_checkpoint, model_type=args.sam_model_type)
def inference_step(self, first_flag: bool, interact_flag: bool, image: np.ndarray, def inference_step(self, first_flag: bool, interact_flag: bool, image: np.ndarray,