sanghoon / pva-faster-rcnn

Demo code for PVANet
https://arxiv.org/abs/1611.08588
Other
651 stars 241 forks source link

Bug in gen_merged_model.py #64

Open Amocy-Wang opened 7 years ago

Amocy-Wang commented 7 years ago

I have found a bug in gen_merged_model.py which i used for model fusion. It is about failure for setting engine to 'CAFFE' when i used this script. I figured out that the return value of layer.convolution_param.kernel_size was always a list rather than a single value. I have fix this and sent pull request.