LCFractal / AIC21-MTMC

🏆The 1st place solution of track3 (City-Scale Multi-Camera Vehicle Tracking) in the NVIDIA AI City Challenge at CVPR 2021 Workshop.
MIT License
128 stars 35 forks source link

nan occurs when running run_mcmt.sh #23

Closed kanonjz closed 2 years ago

kanonjz commented 2 years ago

When I run run_mcmt.sh, the following error occurs. Do you have any suggestions?

starting re_ranking
100%|█████████████████████████████████████████████████████████████████████████████████████| 78/78 [00:00<00:00, 6529.14it/s]
Using totally 0.06S to compute R
100%|█████████████████████████████████████████████████████████████████████████████████████| 78/78 [00:00<00:00, 7608.45it/s]
Using totally 0.10S to compute V-1
Using totally 0.12S to compute V-2
Using totally 0.12S to compute invIndex
100%|█████████████████████████████████████████████████████████████████████████████████████| 39/39 [00:00<00:00, 6944.80it/s]
[[nan nan nan ... nan nan nan]
 [nan nan nan ... nan nan nan]
 [nan nan nan ... nan nan nan]
 ...
 [nan nan nan ... nan nan nan]
 [nan nan nan ... nan nan nan]
 [nan nan nan ... nan nan nan]]
Using totally 0.15S to compute final_distance
[[nan nan nan ... nan nan nan]
 [nan nan nan ... nan nan nan]
 [nan nan nan ... nan nan nan]
 ...
 [nan nan nan ... nan nan nan]
 [nan nan nan ... nan nan nan]
 [nan nan nan ... nan nan nan]]
Traceback (most recent call last):
  File "sub_cluster.py", line 154, in <module>
    clu = get_labels(cfg,cid_tid_dict,cid_tids,score_thr=cfg.SCORE_THR)
  File "sub_cluster.py", line 106, in get_labels
    linkage='complete').fit_predict(1 - sim_matrix)
  File "/home/ubuntu/anaconda3/envs/torch/lib/python3.7/site-packages/sklearn/cluster/_agglomerative.py", line 1054, in fit_predict
    return super().fit_predict(X, y)
  File "/home/ubuntu/anaconda3/envs/torch/lib/python3.7/site-packages/sklearn/base.py", line 736, in fit_predict
    self.fit(X)
  File "/home/ubuntu/anaconda3/envs/torch/lib/python3.7/site-packages/sklearn/cluster/_agglomerative.py", line 917, in fit
    X = self._validate_data(X, ensure_min_samples=2, estimator=self)
  File "/home/ubuntu/anaconda3/envs/torch/lib/python3.7/site-packages/sklearn/base.py", line 566, in _validate_data
    X = check_array(X, **check_params)
  File "/home/ubuntu/anaconda3/envs/torch/lib/python3.7/site-packages/sklearn/utils/validation.py", line 800, in check_array
    _assert_all_finite(array, allow_nan=force_all_finite == "allow-nan")
  File "/home/ubuntu/anaconda3/envs/torch/lib/python3.7/site-packages/sklearn/utils/validation.py", line 116, in _assert_all_finite
    type_err, msg_dtype if msg_dtype is not None else X.dtype
ValueError: Input contains NaN, infinity or a value too large for dtype('float32').
Traceback (most recent call last):
  File "gen_res.py", line 40, in <module>
    map_tid = pickle.load(open('test_cluster.pkl', 'rb'))['cluster']
FileNotFoundError: [Errno 2] No such file or directory: 'test_cluster.pkl'
LCFractal commented 2 years ago

Maybe you generated the wrong distance matrix, please refer to https://github.com/LCFractal/AIC21-MTMC/issues/16#issuecomment-1102496391 Confirm that the program produces the correct "{}_dets_feat.pkl"