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')