Mirrors/Track-Anything
1
0
Fork 0
mirror of https://github.com/gaomingqi/Track-Anything.git synced 2026-05-18 05:05:29 +02:00
Code Issues Packages Projects Releases Wiki Activity

add mask and point painter

Browse Source
This commit is contained in:
gaomingqi
2023-04-13 23:17:17 +08:00
parent 0618c12d63
commit 81009e4447
11 changed files with 193 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tracker/eval.py
View File

@@ -205,10 +205,6 @@ for vid_reader in progressbar(meta_loader, max_value=len(meta_dataset), redirect
# Run the model on this frame
prob = processor.step(rgb, msk, labels, end=(ti==vid_length-1)) # 0, background, >0, objects
# consider prob (only object channels) as prompt to refine segment results
# Upsample to original size if needed
if need_resize:
prob = F.interpolate(prob.unsqueeze(1), shape, mode='bilinear', align_corners=False)[:,0]