ignacio-rocco/detectorch

Error when loading pretrained weights

Opened this issue · 0 comments

Dear @ignacio-rocco,
Thank you for your fantastic work. When I want to run your demo_FPN code, I get this error:

Loading pretrained weights:
Traceback (most recent call last):
  File "SampleRun1.py", line 102, in <module>
    mask_head_type='1up4convs')
  File ".../detectorch/lib/model/detector.py", line 226, in __init__
    self.load_pretrained_weights(detector_pkl_file, model='detector')
  File ".../detectorch/lib/model/detector.py", line 294, in load_pretrained_weights
    caffe_data = pickle.load(f, encoding='latin1')
ValueError: could not convert string to int

Would you please kindly help me to address this error?