|
我使用自己训练好的权重和训练时用的cfg文件转换模型时报错,请问可能是什么原因造成的?错误如下:
--> Building model
Traceback (most recent call last):
File "rknn_transform_512x512.py", line 27, in <module>
rknn.build(do_quantization=False, dataset='./dataset_512x512.txt', pre_compile=False)
File "/home/yy/.local/lib/python3.6/site-packages/rknn/api/rknn.py", line 220, in build
inputs = self.rknn_base.net.get_input_layers()
AttributeError: 'NoneType' object has no attribute 'get_input_layers'
由于darknet没有MobileNet的DW_CONV层,因此我在cfg文件内设置conv层的group以替代DW_CONV层,如下:
[convolutional]
group=64
batch_normalize=1
filters=64
size=3
stride=1
pad=1
activation=relu
但是我看转换模型时生成的.cfg.dat文件里的group_number依然是1。可能是这个原因造成的模型转换失败吗?
|
|