jay-mahadeokar / pynetbuilder

pyNetBuilder is a modular pytonic interface with builtin modules for generating popular caffe prototxt network file definitions.
BSD 2-Clause "Simplified" License
328 stars 140 forks source link

AttributeError: 'module' object has no attribute 'PriorBoxParameter' #12

Closed jahyunkoo closed 7 years ago

jahyunkoo commented 7 years ago

I'm trying to making resnet50-ssd. But I have following problem. Do you have any idea?

python app/ssd/create_ssdnet.py --type Resnet -n 256 -b 3 4 6 3 --no-fc_layers -m bottleneck --extra_blocks 3 3 --extra_num_outputs 2048 2048 --mbox_source_layers relu_stage1_block3 relu_stage2_block5 relu_stage3_block2 relu_stage4_block2 relu_stage5_block2 pool_last --extra_layer_attach pool -c 21 -o ./

error message Resnet [30.0, 60.0, 114.0, 168.0, 222.0, 276.0] [[], 114.0, 168.0, 222.0, 276.0, 330.0] [[2], [2, 3], [2, 3], [2, 3], [2, 3], [2, 3]] [20, 20, 20, -1, -1, -1] Traceback (most recent call last): File "app/ssd/create_ssdnet.py", line 56, in netspec = get_resnet_ssdnet(res_params) File "netbuilder/nets/ssdnet.py", line 166, in get_resnet_ssdnet MBoxAssembleLego(assemble_params).attach(netspec, [netspec['label']]) File "netbuilder/lego/ssd.py", line 260, in attach code_type = P.PriorBox.CENTER_SIZE File "/home/ubuntu/caffe/python/caffe/net_spec.py", line 220, in getattr return getattr(getattr(caffe_pb2, name + 'Parameter'), param_name) AttributeError: 'module' object has no attribute 'PriorBoxParameter'