default parameter for template image segment for the first test - watchtower

This commit is contained in:
memoryunreal
2023-04-14 02:27:39 +00:00
parent d3f737ede5
commit 2fb43bf75e
4 changed files with 93 additions and 31 deletions

View File

@@ -15,7 +15,7 @@ from dataset.range_transform import im_normalization
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