search

ignacio-rocco / detectorch

Detectorch - detectron for PyTorch
Other
558 stars 72 forks source link

Error when loading pretrained weights #13

Open ahkarami opened 6 years ago

ahkarami commented 6 years ago

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?