This commit is contained in:
gaomingqi
2023-04-14 12:37:46 +08:00
5 changed files with 136 additions and 48 deletions

View File

@@ -21,7 +21,7 @@ from tools.painter import mask_painter
class BaseTracker:
def __init__(self, device, xmem_checkpoint) -> None:
def __init__(self, xmem_checkpoint, device) -> None:
"""
device: model device
xmem_checkpoint: checkpoint of XMem model