search

samihormi / Multi-Camera-Person-Tracking-and-Re-Identification

Simple model to Track and Re-identify individuals in different cameras/videos.(Yolov3 & Yolov4)
GNU General Public License v3.0
203 stars 68 forks source link

Pretrained model weights cannot be loaded #16

Open yuxin212 opened 1 year ago

yuxin212 commented 1 year ago

Hi,

The pretrained Torchreid models you provided cannot be loaded.

If you comment out L323 in demo.py, there should be a warning indicating this:

/Multi-Camera-Person-Tracking-and-Re-Identification/torchreid/utils/torchtools.py:299: UserWarning: The pretrained weights 
"model_data/models/model.pth" cannot be loaded, please check the key names manually (** ignored and continue **)

The 2 model weights you provided cannot fit any of the available Torchreid models.

Could you please check and update the Torchreid model weights? Thanks!

dranhclub commented 1 year ago

When I used the first provided model weight, I met the same error. The problem was solved when I used the second model weight. Note that you need to edit the model path in reid.py to 'model_data/models/model.pth.tar-60' if you use the second one.