预训练系数用的
ctw1500_res50_pretrain_ic17.pth
环境如repo要求:
python2.7
....
报错如下:
Traceback (most recent call last):
File "/home/hong/PJTs/OCR/PSENet/test_ctw1500.py", line 219, in
test(args)
File "/home/hong/PJTs/OCR/PSENet/test_ctw1500.py", line 88, in test
model = models.resnet50(pretrained=True, num_classes=7, scale=args.scale)
File "/home/hong/PJTs/OCR/PSENet/models/fpn_resnet.py", line 269, in resnet50
pretrained_model = model_zoo.load_url(model_urls['resnet50'])
File "/home/hong/anaconda3/envs/Psenet/lib/python2.7/site-packages/torch/hub.py", line 506, in load_state_dict_from_url
return torch.load(cached_file, map_location=map_location)
File "/home/hong/anaconda3/envs/Psenet/lib/python2.7/site-packages/torch/serialization.py", line 529, in load
return _legacy_load(opened_file, map_location, pickle_module, pickle_load_args)
File "/home/hong/anaconda3/envs/Psenet/lib/python2.7/site-packages/torch/serialization.py", line 692, in _legacy_load
magic_number = pickle_module.load(f, pickle_load_args)
cPickle.UnpicklingError: bad pickle data
预训练系数用的 ctw1500_res50_pretrain_ic17.pth 环境如repo要求: python2.7 .... 报错如下: Traceback (most recent call last): File "/home/hong/PJTs/OCR/PSENet/test_ctw1500.py", line 219, in
test(args)
File "/home/hong/PJTs/OCR/PSENet/test_ctw1500.py", line 88, in test
model = models.resnet50(pretrained=True, num_classes=7, scale=args.scale)
File "/home/hong/PJTs/OCR/PSENet/models/fpn_resnet.py", line 269, in resnet50
pretrained_model = model_zoo.load_url(model_urls['resnet50'])
File "/home/hong/anaconda3/envs/Psenet/lib/python2.7/site-packages/torch/hub.py", line 506, in load_state_dict_from_url
return torch.load(cached_file, map_location=map_location)
File "/home/hong/anaconda3/envs/Psenet/lib/python2.7/site-packages/torch/serialization.py", line 529, in load
return _legacy_load(opened_file, map_location, pickle_module, pickle_load_args)
File "/home/hong/anaconda3/envs/Psenet/lib/python2.7/site-packages/torch/serialization.py", line 692, in _legacy_load
magic_number = pickle_module.load(f, pickle_load_args)
cPickle.UnpicklingError: bad pickle data