Storife / RANet

RANet: Ranking Attention Network for Fast Video Object Segmentation (VOS), ICCV2019

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Unpickling error

sourabhswain opened this issue · comments

Hi, I followed the instructions and am trying to generate the predictions for DAVIS17testdev using the RANet_video_multi.pth. However, I am facing the following error :

Environment :
Python - 3.6.9
Torchvision version - 0.2.0

Can you please tell me how to resolve this ?

===> Setting ...... ../models/exists using device ID: [0] ===> Building model Single-object mode save root = ../predictions/RANet_Video_17test_dev Change to multi-object mode Traceback (most recent call last): File "RANet.py", line 104, in <module> predict_SVOS(params='RANet_video_multi.pth', dataset='17test_dev', save_root='../predictions/RANet_Video_17test_dev') File "RANet.py", line 83, in predict_SVOS checkpoint_load(opt.workfolder + params, model) File "/home/swain/Documents/VOS/Methods/RANet/codes/RANet_lib/RANet_lib.py", line 38, in checkpoint_load checkpoint = torch.load(fpath) File "/home/swain/anaconda3/envs/vos/lib/python3.6/site-packages/torch/serialization.py", line 426, in load return _load(f, map_location, pickle_module, **pickle_load_args) File "/home/swain/anaconda3/envs/vos/lib/python3.6/site-packages/torch/serialization.py", line 603, in _load magic_number = pickle_module.load(f, **pickle_load_args) _pickle.UnpicklingError: invalid load key, '<'.