Sentient07/HeadHunter--T

regression_thresh needs to be lower than 0.1

Closed this issue · 1 comments

I put regression_thresh to 0 in the config_reproduce.yaml file, otherwise, I was ending up with this error if I run run_mot.py.
RuntimeError: boxes should be a 2d tensor, got 1D
Do you have an explanation for this ?

The problem happened because I didn't install the packages versions according to your .yaml file.

For the sake of completion, these packages need to be adjusted/added:

- torch==1.8.1+cu111 torchvision==0.9.1+cu111 torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html
- albumentations==0.5.2
- munkres
- pyyaml

Make sure to not install a newer version of albumentations, because ToTensor has been removed from newer versions.
Some from scipy.misc import imread, imsave need to be replaced by matplotlib.pyplot import imread, imsave