Update app.py

please input args.port and args.device by yourself
This commit is contained in:
Zhe Li
2023-04-19 00:35:02 +08:00
committed by GitHub
parent 5ed65a6ee6
commit 78ff72de87

6
app.py
View File

@@ -193,9 +193,9 @@ xmem_checkpoint = download_checkpoint(xmem_checkpoint_url, folder, xmem_checkpoi
# args, defined in track_anything.py # args, defined in track_anything.py
args = parse_augment() args = parse_augment()
args.port = 12212 # args.port = 12212
args.device = "cuda:4" # args.device = "cuda:4"
args.mask_save = True # args.mask_save = True
model = TrackingAnything(SAM_checkpoint, xmem_checkpoint, args) model = TrackingAnything(SAM_checkpoint, xmem_checkpoint, args)