Merge branch 'master' of https://github.com/YinqiangZhang/MOT_MUST2

YinqiangZhang · Oct 1, 2019 · 8e75f8b · 8e75f8b
2 parents d1a8343 + 32b569f
commit 8e75f8b
Show file tree

Hide file tree

Showing 7 changed files with 3 additions and 984 deletions.
diff --git a/detector/yolov3/LICENSE b/detector/yolov3/LICENSE
diff --git a/detector/yolov3/README.md b/detector/yolov3/README.md
diff --git a/detector/yolov3/requirements.txt b/detector/yolov3/requirements.txt
diff --git a/detector/yolov3/run.sh b/detector/yolov3/run.sh
diff --git a/pysot/ReID/LICENSE b/pysot/ReID/LICENSE
diff --git a/pysot/ReID/README.md b/pysot/ReID/README.md
diff --git a/pysot/mot_zj/MUST_ASSO/MUST_asso_model.py b/pysot/mot_zj/MUST_ASSO/MUST_asso_model.py
@@ -32,15 +32,15 @@ def __init__(self):
         checkpoint = load_checkpoint(os.path.join(os.getcwd(), 'weights', 'checkpoint.pth.tar'))
         self.model.load_state_dict(checkpoint['state_dict'])
         self.model.eval()
-        self.frame_root = os.path.join('.\\result\\img')
-        self.tracklet_root = "pysot\\img_traj"
+        self.frame_root = os.path.join(os.getcwd(), 'result', 'img')
+        self.tracklet_root = os.path.join("pysot","img_traj")
         if torch.cuda.is_available():
             self.model = self.model.cuda()
         print('load weights done!')
 
     def __call__(self, bboxes_asso, seq_name, frame, id_num):
         img_trajs = []
-        frame_path = os.path.join(self.frame_root, seq_name, "img1\\{:06d}.jpg".format(frame))
+        frame_path = os.path.join(self.frame_root, seq_name, "img1", "{:06d}.jpg".format(frame))
         traj_dir = os.path.join(self.tracklet_root, seq_name, str(id_num))
 
         img_frame = cv2.imread(frame_path) # the whole frame image needed