Open liguoyu666 opened 4 years ago
I found that in the conversion process,
`
param_size = np.product(np.array(param_shape))
print(param_name, param_shape, param_size)
if param_name == '110_convolutional_conv_weights':
print(len(self.weights_file.read(param_size * 4)))
param_data = np.ndarray(
shape=param_shape,
dtype='float32',
buffer=self.weights_file.read(param_size * 4))
print(param_data.shape)
print('------')`
when converting layer ‘110_convolutional_conv_weights’, the weight was not enough
110_convolutional_conv_weights [256, 128, 3, 3] 294912
661500
Traceback (most recent call last):
File "spp_weight_to_onnx.py", line 726, in
When I convert yolov3-spp to onnx, it reported:
File "/home/lgy/PycharmProjects/TensorRT_yolo3_module/spp_weight_to_onnx.py", line 296, in _load_one_param_type buffer=self.weights_file.read(param_size * 4)) TypeError: buffer is too small for requested array
I am sure yolov3-spp.cfg and yolov3-spp.weights are matched.
could you help me?