dump_tensorflow_weights.py with error

[he-chen]

i run the following command :python3 dump_tensorflow_weights.py and get this error: Traceback (most recent call last): File "dump_tensorflow_weights.py", line 95, in tmp = caffe_weights.reshape(boxes, -1).copy() TypeError: 'float' object cannot be interpreted as an integer

i fix it by change the code at line 71

if output_name.find('BoxEncodingPredictor') != -1: boxes = caffe_weights.shape[0] / 4 elif output_name.find('ClassPredictor') != -1: boxes = caffe_weights.shape[0] / 91

to

if output_name.find('BoxEncodingPredictor') != -1: boxes = caffe_weights.shape[0] // 4 elif output_name.find('ClassPredictor') != -1: boxes = caffe_weights.shape[0] // 91

and i wonder if it is ok or not ?

[zhanghanbin3159]

it ok! this script is write by python2, if you use python3,should change '/' to '//'

[chuanqi305]

@he-chen Thank you very much, I have merged your code to my project.