我的是直接转换pt文件。
D RKNN output shape(batchnormalize): (0 16 16 320)
D Process convolution_at_input136.1_36 ...
D RKNN output shape(convolution): (0 16 16 1280)
D Process batch_norm_at_input137.1_35 ...
D RKNN output shape(batchnormalize): (0 16 16 1280)
D Process hardtanh__at_input138.1_34 ...
D RKNN output shape(relun): (0 16 16 1280)
D Process convolution_at_input139.1_33 ...
D RKNN output shape(convolution): (0 8 8 1280)
D Process batch_norm_at_input140.1_32 ...
D RKNN output shape(batchnormalize): (0 8 8 1280)
D Process relu__at_input141.1_31 ...
D RKNN output shape(relu): (0 8 8 1280)
D Process convolution_at_input142.1_30 ...
D RKNN output shape(convolution): (0 4 4 16)
D Process batch_norm_at_input143.1_29 ...
D RKNN output shape(batchnormalize): (0 4 4 16)
D Process relu__at_input144.1_28 ...
D RKNN output shape(relu): (0 4 4 16)
D Process convolution_at_input145.1_27 ...
D RKNN output shape(convolution): (0 2 2 16)
D Process batch_norm_at_input146.1_26 ...
D RKNN output shape(batchnormalize): (0 2 2 16)
D Process relu__at_input147.1_24 ...
D RKNN output shape(relu): (0 2 2 16)
D Process convolution_at_input148.1_18 ...
D RKNN output shape(convolution): (0 2 2 64)
D Process relu__at_input149.1_12 ...
D RKNN output shape(relu): (0 2 2 64)
D Process convolution_at_hm_x.1_6 ...
D Process output_of_convolution_at_2269_5 ...
D RKNN output shape(output): (0 2 2 2)
I Build mobilenet1 complete.
D Optimizing network with force_1d_tensor, swapper, merge_layer, auto_fill_bn, resize_nearest_transformer, auto_fill_multiply, merge_avgpool_conv1x1, auto_fill_zero_bias, proposal_opt_import
D Merge ['convolution_at_input145.1_27', 'batch_norm_at_input146.1_26'] (convolution)
D Merge ['convolution_at_input142.1_30', 'batch_norm_at_input143.1_29'] (convolution)
D Merge ['convolution_at_input139.1_33', 'batch_norm_at_input140.1_32'] (convolution)
E Catch exception when loading pytorch model: /Users/gyq/DeepLearning/CenterNet_in_mac/mobilenet1.pt!
E Traceback (most recent call last):
E File "rknn/api/rknn_base.py", line 567, in rknn.api.rknn_base.RKNNBase.load_pytorch
Load pytorch model failed!
E File "rknn/base/RKNNlib/app/importer/import_pytorch.py", line 121, in rknn.base.RKNNlib.app.importer.import_pytorch.ImportPytorch.run
E File "rknn/base/RKNNlib/app/helper/mergehelper.py", line 155, in rknn.base.RKNNlib.app.helper.mergehelper.MergeHelper.merge
E File "rknn/base/RKNNlib/optimize/optimizer.py", line 299, in rknn.base.RKNNlib.optimize.optimizer.Optimizer.apply
E File "rknn/base/RKNNlib/optimize/rules/merge_layer.py", line 81, in rknn.base.RKNNlib.optimize.rules.merge_layer.MergeLayer.apply
E File "rknn/base/RKNNlib/optimize/rules/merge_layer.py", line 105, in rknn.base.RKNNlib.optimize.rules.merge_layer.MergeLayer._loop
E File "rknn/base/RKNNlib/optimize/rules/merge_layer_ext_proc.py", line 86, in rknn.base.RKNNlib.optimize.rules.merge_layer_ext_proc.m_l1_bn
E ValueError: operands could not be broadcast together with shapes (4,4,16,1280) (16,)