Sorry,why i got the wrong predict pose ,when i used your pretrained model?

[kevin860213]

I got the image from your pretrained model. Do you have any ideas with this problem? thank you and your project

[pengsida]

The pretrained model should be put at $ROOT/data/model/pvnet/cat/199.pth.
And run python run.py --type visualize --cfg_file configs/linemod.yaml model cat.

[kevin860213]

sorry, i didn't create the folder of pvnet , i directly put the model into data.
thank you , it was sloved

[pengsida]

Enjoy this code!

[pengsida]

If possible, make more people to know this project.
Thank you!

[kevin860213]

OK, I will.
Thanks!

[jinglan0829]

I put the pretrained model at $ROOT/data/model/pvnet/cat/199.pth.
But I have error with :
Traceback (most recent call last):
File "run.py", line 198, in
globals()'run_'+args.type
File "run.py", line 90, in run_visualize
load_network(network, cfg.model_dir, resume=cfg.resume, epoch=cfg.test.epoch)
File "/home/jinglan/clean-pvnet-master/lib/utils/net_utils.py", line 318, in load_network
pretrained_model = torch.load(os.path.join(model_dir, '{}.pth'.format(pth)))
File "/usr/local/lib/python3.5/dist-packages/torch/serialization.py", line 387, in load
return _load(f, map_location, pickle_module, **pickle_load_args)
File "/usr/local/lib/python3.5/dist-packages/torch/serialization.py", line 564, in _load
magic_number = pickle_module.load(f, **pickle_load_args)
_pickle.UnpicklingError: invalid load key, ' '.

[pengsida]

The pth file is broken.

[jinglan0829]

I'll check it. Thanks!