TRACKLET_SIZE = config["TRACKLET"]['TRACKLET_SIZE']
# REID PARAMETERS
REID_INSTANCE_SIZE = config["REID"]['REID_INSTANCE_SIZE']
REID_BACKBONE = config["REID"]['BACKBONE']
IOU_THRESHOLD = config["EVAL"]['IOU_THRESHOLD']
# DETECTION PARAMETERS
lasot_dir = lasot_dir + "/*/"
folders = glob.glob(lasot_dir)
# print(folders[0].split('\\')[-2])
tracker = Tracking(
config='tracking/experiments/siamrpn_r50_l234_dwxcorr/config.yaml',
snapshot='tracking/experiments/siamrpn_r50_l234_dwxcorr/model.pth')
detector = Detection(
config="./detectron2/configs/COCO-InstanceSegmentation/small.yaml",
model=
"detectron2://COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x/137849600/model_final_f10217.pkl"
)
reid_module = REID(model=REID_BACKBONE)
tracklet = Tracklet(TRACKLET_SIZE)
running_stats = RunningStats()
import sys
import os
import cv2
import torch
import numpy as np
from glob import glob
sys.path.append('./tracking/')
print(sys.path)
from tracking.sot import Tracking
from reid import REID
from detection import Detection
tracker = Tracking(
config='tracking/experiments/siamrpn_r50_l234_dwxcorr/config.yaml',
snapshot='tracking/experiments/siamrpn_r50_l234_dwxcorr/model.pth')
detector = Detection(
config="./detectron2/configs/COCO-InstanceSegmentation/small.yaml",
model=
"detectron2://COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x/137849600/model_final_f10217.pkl"
)
reid_module = REID(model='resnet18')