jiaweihe1996 / GMTracker

Official PyTorch implementation of "Learnable Graph Matching: Incorporating Graph Partitioning with Deep Feature Learning for Multiple Object Tracking" (CVPR 2021).
GNU General Public License v3.0
112 stars 23 forks source link

Problems encountered when running gmtracker_app.py #8

Closed esflfei closed 3 years ago

esflfei commented 3 years ago

I have finished 1~4 steps in 'Getting Started'. However, when I try to run the first example code in step 5:

python gmtracker_app.py --sequence_dir data/MOT17/test/MOT17-01 --detection_file npy/npytest_tracktor/MOT17-01-DPM.npy --checkpoint_dir experiments/static/params/0001 --max_age 100 --reid_thr 0.6 --output_file results/test/MOT17-01-DPM.txt

An error occors:

torch.Size([2171, 512]) Loading model parameters from experiments/static/params/0001/params.pt torch.Size([2171, 512]) Traceback (most recent call last): File "gmtracker_app.py", line 285, in args.max_age, args.n_init, args.reid_thr, args.checkpoint_dir) File "gmtracker_app.py", line 205, in run seq_info = gather_sequence_info(sequence_dir, new_npy) File "gmtracker_app.py", line 46, in gather_sequence_info for f in os.listdir(image_dir)} FileNotFoundError: [Errno 2] No such file or directory: 'data/MOT17/test/MOT17-01/img1'

The error also occurs when I run the second example code in step 5. I guess the error happens because of the wrong address of --sequence_dir, so I run the following code (only sequence_dir changes):

python gmtracker_app.py --sequence_dir data/MOT17/test/MOT17-01-DPM --detection_file npy/npytest_tracktor/MOT17-01-DPM.npy --checkpoint_dir experiments/static/params/0001 --max_age 100 --reid_thr 0.6 --output_file results/test/MOT17-01-DPM.txt

Another error occurs:

torch.Size([2171, 512]) Loading model parameters from experiments/static/params/0001/params.pt torch.Size([2171, 512]) Processing MOT17-01-DPM frame 00001 Nothing to match at this level Processing MOT17-01-DPM frame 00002 Nothing to match at this level Processing MOT17-01-DPM frame 00003 Traceback (most recent call last): File "gmtracker_app.py", line 285, in args.max_age, args.n_init, args.reid_thr, args.checkpoint_dir) File "gmtracker_app.py", line 239, in run frame_callback(frame_idx) File "gmtracker_app.py", line 226, in frame_callback tracker.update(detections, seq_info["sequence_name"], frame_idx, checkpoint_dir) File "/mnt/nas/home/jiangshaofei/GMTracker/utils/tracker.py", line 119, in update self._match(detections,video,frame,checkpoint_dir) File "/mnt/nas/home/jiangshaofei/GMTracker/utils/tracker.py", line 149, in _match self.max_age, self.tracks, detections, confirmed_tracks,reid_thr=self.reid_thr,seq_name=video,ckp_dir=checkpoint_dir) File "/mnt/nas/home/jiangshaofei/GMTracker/GMMOT/assignment.py", line 143, in graph_matching track_indices_l, unmatched_detections,reid_thr,seq_name,ckp_dir) File "/mnt/nas/home/jiangshaofei/GMTracker/GMMOT/assignment.py", line 95, in quadratic_matching indices, thr_flag = graphnet.forward(data1, data2, kf_gate, reid_thr, iou, start_src, end_src, start_tgt, end_tgt, seq_name, inverse_flag=False) File "/mnt/nas/home/jiangshaofei/GMTracker/GMMOT/gmatching.py", line 30, in forward m_emb1 = torch.bmm(Mp0, emb2) UnboundLocalError: local variable 'Mp0' referenced before assignment

I have no idea... Can you help me?

jiaweihe1996 commented 3 years ago

Please rename the data/MOT17/test/MOT17-01-DPM to data/MOT17/test/MOT17-01 and follow the data files' structure in step 3, because the folder name MOT17-01-DPM is not in https://github.com/jiaweihe1996/GMTracker/blob/c3cc6d005214d5e973c718019a6b4a3a1fc517b7/utils/config.py#L43-L44

esflfei commented 3 years ago

Oh, I missed it. It works well now. Thanks a lot! Here comes an additional question: what's the difference between 'DPM', 'FRCNN' and 'SDP' at the end of a folder's name? In other words, what does the three mean?

jiaweihe1996 commented 3 years ago

These are three different public detection methods provided by the MOTChallenge benchmark.

esflfei commented 3 years ago

Oh, I see. Thanks a lot!