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

hugging space badge -- li

Browse Source
This commit is contained in:
memoryunreal
2023-04-26 05:04:57 +00:00
parent f5df95ed9a
commit 3f91190b8e
4 changed files with 23 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
track_anything.py
View File

@@ -10,9 +10,12 @@ import argparse
class TrackingAnything():
def __init__(self, sam_checkpoint, xmem_checkpoint, e2fgvi_checkpoint, args):
self.args = args
self.samcontroler = SamControler(sam_checkpoint, args.sam_model_type, args.device)
self.xmem = BaseTracker(xmem_checkpoint, device=args.device)
self.baseinpainter = BaseInpainter(e2fgvi_checkpoint, args.device)
self.sam_checkpoint = sam_checkpoint
self.xmem_checkpoint = xmem_checkpoint
self.e2fgvi_checkpoint = e2fgvi_checkpoint
self.samcontroler = SamControler(self.sam_checkpoint, args.sam_model_type, args.device)
self.xmem = BaseTracker(self.xmem_checkpoint, device=args.device)
self.baseinpainter = BaseInpainter(self.e2fgvi_checkpoint, args.device)
# def inference_step(self, first_flag: bool, interact_flag: bool, image: np.ndarray,
# same_image_flag: bool, points:np.ndarray, labels: np.ndarray, logits: np.ndarray=None, multimask=True):
# if first_flag: