pointnet.pytorch icon indicating copy to clipboard operation
pointnet.pytorch copied to clipboard

KeyError: 'conv4.weight'

Open HeBangYan opened this issue 2 years ago • 0 comments

I ran this command python utils/show_seg.py and reported the following error:

Namespace(class_choice='Chair', dataset='./shapenetcore_partanno_segmentation_benchmark_v0', idx=0, model='./seg/seg_model_Chair_29.pth')
{'Chair': 0}
{'Airplane': 4, 'Bag': 2, 'Cap': 2, 'Car': 4, 'Chair': 4, 'Earphone': 3, 'Guitar': 3, 'Knife': 2, 'Lamp': 4, 'Laptop': 2, 'Motorbike': 6, 'Mug': 2, 'Pistol': 3, 'Rocket': 3, 'Skateboard': 3, 'Table': 3} 4
model 0/704
torch.Size([2500, 3]) torch.Size([2500])
Traceback (most recent call last):
  File "utils/show_seg.py", line 45, in <module>
    classifier = PointNetDenseCls(k= state_dict['conv4.weight'].size()[0])
KeyError: 'conv4.weight'

What should I do? Looking forward to your reply.

HeBangYan avatar Dec 31 '21 06:12 HeBangYan