Toybrick

标题: 转化keras框架的yolov3模型出现问题 [打印本页]

作者: jinhao 时间: 2021-3-30 13:53
标题: 转化keras框架的yolov3模型出现问题
本人想尝试转化一个keras框架的yolov3模型，已经在PC上测试过该.h5文件可以正常使用，完成目标检测任务。但是在ubuntu18虚拟环境将.h5文件转化为.rknn文件会出现模型建立失败的错误，初步判断是权重文件和转化器不匹配，请问该解决这个问题

这是我下载的权重文件的地址：https://pjreddie.com/media/files/yolov3.weights

(tf1) kjh@ubuntu:~/桌面/yolo3-keras-master$ python testrknn.py
--> Config model
done
--> Loading model
W:tensorflow:No training configuration found in save file: the model was *not* compiled. Compile it manually.
__________________________________________________________________________________________________
Layer (type)                   Output Shape       Param #    Connected to
==================================================================================================
input_1 (InputLayer)          (None, None, None, 3 0
__________________________________________________________________________________________________
conv2d_1 (Conv2D)             (None, None, None, 3 864       input_1[0][0]
__________________________________________________________________________________________________
batch_normalization_1 (BatchNor (None, None, None, 3 128       conv2d_1[0][0]
__________________________________________________________________________________________________
leaky_re_lu_1 (LeakyReLU)    (None, None, None, 3 0          batch_normalization_1[0][0]
__________________________________________________________________________________________________
zero_padding2d_1 (ZeroPadding2D (None, None, None, 3 0          leaky_re_lu_1[0][0]
__________________________________________________________________________________________________
conv2d_2 (Conv2D)             (None, None, None, 6 18432    zero_padding2d_1[0][0]
__________________________________________________________________________________________________
batch_normalization_2 (BatchNor (None, None, None, 6 256       conv2d_2[0][0]
__________________________________________________________________________________________________
leaky_re_lu_2 (LeakyReLU)    (None, None, None, 6 0          batch_normalization_2[0][0]
__________________________________________________________________________________________________
conv2d_3 (Conv2D)             (None, None, None, 3 2048       leaky_re_lu_2[0][0]
__________________________________________________________________________________________________
batch_normalization_3 (BatchNor (None, None, None, 3 128       conv2d_3[0][0]
__________________________________________________________________________________________________
leaky_re_lu_3 (LeakyReLU)    (None, None, None, 3 0          batch_normalization_3[0][0]
__________________________________________________________________________________________________
conv2d_4 (Conv2D)             (None, None, None, 6 18432    leaky_re_lu_3[0][0]
__________________________________________________________________________________________________
batch_normalization_4 (BatchNor (None, None, None, 6 256       conv2d_4[0][0]
__________________________________________________________________________________________________
leaky_re_lu_4 (LeakyReLU)    (None, None, None, 6 0          batch_normalization_4[0][0]
__________________________________________________________________________________________________
add_1 (Add)                   (None, None, None, 6 0          leaky_re_lu_2[0][0]
                                                               leaky_re_lu_4[0][0]
__________________________________________________________________________________________________
zero_padding2d_2 (ZeroPadding2D (None, None, None, 6 0          add_1[0][0]
__________________________________________________________________________________________________
conv2d_5 (Conv2D)             (None, None, None, 1 73728    zero_padding2d_2[0][0]
__________________________________________________________________________________________________
batch_normalization_5 (BatchNor (None, None, None, 1 512       conv2d_5[0][0]
__________________________________________________________________________________________________
leaky_re_lu_5 (LeakyReLU)    (None, None, None, 1 0          batch_normalization_5[0][0]
__________________________________________________________________________________________________
conv2d_6 (Conv2D)             (None, None, None, 6 8192       leaky_re_lu_5[0][0]
__________________________________________________________________________________________________
batch_normalization_6 (BatchNor (None, None, None, 6 256       conv2d_6[0][0]
__________________________________________________________________________________________________
leaky_re_lu_6 (LeakyReLU)    (None, None, None, 6 0          batch_normalization_6[0][0]
__________________________________________________________________________________________________
conv2d_7 (Conv2D)             (None, None, None, 1 73728    leaky_re_lu_6[0][0]
__________________________________________________________________________________________________
batch_normalization_7 (BatchNor (None, None, None, 1 512       conv2d_7[0][0]
__________________________________________________________________________________________________
leaky_re_lu_7 (LeakyReLU)    (None, None, None, 1 0          batch_normalization_7[0][0]
__________________________________________________________________________________________________
add_2 (Add)                   (None, None, None, 1 0          leaky_re_lu_5[0][0]
                                                               leaky_re_lu_7[0][0]
__________________________________________________________________________________________________
conv2d_8 (Conv2D)             (None, None, None, 6 8192       add_2[0][0]
__________________________________________________________________________________________________
batch_normalization_8 (BatchNor (None, None, None, 6 256       conv2d_8[0][0]
__________________________________________________________________________________________________
leaky_re_lu_8 (LeakyReLU)    (None, None, None, 6 0          batch_normalization_8[0][0]
__________________________________________________________________________________________________
conv2d_9 (Conv2D)             (None, None, None, 1 73728    leaky_re_lu_8[0][0]
__________________________________________________________________________________________________
batch_normalization_9 (BatchNor (None, None, None, 1 512       conv2d_9[0][0]
__________________________________________________________________________________________________
leaky_re_lu_9 (LeakyReLU)    (None, None, None, 1 0          batch_normalization_9[0][0]
__________________________________________________________________________________________________
add_3 (Add)                   (None, None, None, 1 0          add_2[0][0]
                                                               leaky_re_lu_9[0][0]
__________________________________________________________________________________________________
zero_padding2d_3 (ZeroPadding2D (None, None, None, 1 0          add_3[0][0]
__________________________________________________________________________________________________
conv2d_10 (Conv2D)             (None, None, None, 2 294912    zero_padding2d_3[0][0]
__________________________________________________________________________________________________
batch_normalization_10 (BatchNo (None, None, None, 2 1024       conv2d_10[0][0]
__________________________________________________________________________________________________
leaky_re_lu_10 (LeakyReLU)    (None, None, None, 2 0          batch_normalization_10[0][0]
__________________________________________________________________________________________________
conv2d_11 (Conv2D)             (None, None, None, 1 32768    leaky_re_lu_10[0][0]
__________________________________________________________________________________________________
batch_normalization_11 (BatchNo (None, None, None, 1 512       conv2d_11[0][0]
__________________________________________________________________________________________________
leaky_re_lu_11 (LeakyReLU)    (None, None, None, 1 0          batch_normalization_11[0][0]
__________________________________________________________________________________________________
conv2d_12 (Conv2D)             (None, None, None, 2 294912    leaky_re_lu_11[0][0]
__________________________________________________________________________________________________
batch_normalization_12 (BatchNo (None, None, None, 2 1024       conv2d_12[0][0]
__________________________________________________________________________________________________
leaky_re_lu_12 (LeakyReLU)    (None, None, None, 2 0          batch_normalization_12[0][0]
__________________________________________________________________________________________________
add_4 (Add)                   (None, None, None, 2 0          leaky_re_lu_10[0][0]
                                                               leaky_re_lu_12[0][0]
__________________________________________________________________________________________________
conv2d_13 (Conv2D)             (None, None, None, 1 32768    add_4[0][0]
__________________________________________________________________________________________________
batch_normalization_13 (BatchNo (None, None, None, 1 512       conv2d_13[0][0]
__________________________________________________________________________________________________
leaky_re_lu_13 (LeakyReLU)    (None, None, None, 1 0          batch_normalization_13[0][0]
__________________________________________________________________________________________________
conv2d_14 (Conv2D)             (None, None, None, 2 294912    leaky_re_lu_13[0][0]
__________________________________________________________________________________________________
batch_normalization_14 (BatchNo (None, None, None, 2 1024       conv2d_14[0][0]
__________________________________________________________________________________________________
leaky_re_lu_14 (LeakyReLU)    (None, None, None, 2 0          batch_normalization_14[0][0]
__________________________________________________________________________________________________
add_5 (Add)                   (None, None, None, 2 0          add_4[0][0]
                                                               leaky_re_lu_14[0][0]
__________________________________________________________________________________________________
conv2d_15 (Conv2D)             (None, None, None, 1 32768    add_5[0][0]
__________________________________________________________________________________________________
batch_normalization_15 (BatchNo (None, None, None, 1 512       conv2d_15[0][0]
__________________________________________________________________________________________________
leaky_re_lu_15 (LeakyReLU)    (None, None, None, 1 0          batch_normalization_15[0][0]
__________________________________________________________________________________________________
conv2d_16 (Conv2D)             (None, None, None, 2 294912    leaky_re_lu_15[0][0]
__________________________________________________________________________________________________
batch_normalization_16 (BatchNo (None, None, None, 2 1024       conv2d_16[0][0]
__________________________________________________________________________________________________
leaky_re_lu_16 (LeakyReLU)    (None, None, None, 2 0          batch_normalization_16[0][0]
__________________________________________________________________________________________________
add_6 (Add)                   (None, None, None, 2 0          add_5[0][0]
                                                               leaky_re_lu_16[0][0]
__________________________________________________________________________________________________
conv2d_17 (Conv2D)             (None, None, None, 1 32768    add_6[0][0]
__________________________________________________________________________________________________
batch_normalization_17 (BatchNo (None, None, None, 1 512       conv2d_17[0][0]
__________________________________________________________________________________________________
leaky_re_lu_17 (LeakyReLU)    (None, None, None, 1 0          batch_normalization_17[0][0]
__________________________________________________________________________________________________
conv2d_18 (Conv2D)             (None, None, None, 2 294912    leaky_re_lu_17[0][0]
__________________________________________________________________________________________________
batch_normalization_18 (BatchNo (None, None, None, 2 1024       conv2d_18[0][0]
__________________________________________________________________________________________________
leaky_re_lu_18 (LeakyReLU)    (None, None, None, 2 0          batch_normalization_18[0][0]
__________________________________________________________________________________________________
add_7 (Add)                   (None, None, None, 2 0          add_6[0][0]
                                                               leaky_re_lu_18[0][0]
__________________________________________________________________________________________________
conv2d_19 (Conv2D)             (None, None, None, 1 32768    add_7[0][0]
__________________________________________________________________________________________________
batch_normalization_19 (BatchNo (None, None, None, 1 512       conv2d_19[0][0]
__________________________________________________________________________________________________
leaky_re_lu_19 (LeakyReLU)    (None, None, None, 1 0          batch_normalization_19[0][0]
__________________________________________________________________________________________________
conv2d_20 (Conv2D)             (None, None, None, 2 294912    leaky_re_lu_19[0][0]
__________________________________________________________________________________________________
batch_normalization_20 (BatchNo (None, None, None, 2 1024       conv2d_20[0][0]
__________________________________________________________________________________________________
leaky_re_lu_20 (LeakyReLU)    (None, None, None, 2 0          batch_normalization_20[0][0]
__________________________________________________________________________________________________
add_8 (Add)                   (None, None, None, 2 0          add_7[0][0]
                                                               leaky_re_lu_20[0][0]
__________________________________________________________________________________________________
conv2d_21 (Conv2D)             (None, None, None, 1 32768    add_8[0][0]
__________________________________________________________________________________________________
batch_normalization_21 (BatchNo (None, None, None, 1 512       conv2d_21[0][0]
__________________________________________________________________________________________________
leaky_re_lu_21 (LeakyReLU)    (None, None, None, 1 0          batch_normalization_21[0][0]
__________________________________________________________________________________________________
conv2d_22 (Conv2D)             (None, None, None, 2 294912    leaky_re_lu_21[0][0]
__________________________________________________________________________________________________
batch_normalization_22 (BatchNo (None, None, None, 2 1024       conv2d_22[0][0]
__________________________________________________________________________________________________
leaky_re_lu_22 (LeakyReLU)    (None, None, None, 2 0          batch_normalization_22[0][0]
__________________________________________________________________________________________________
add_9 (Add)                   (None, None, None, 2 0          add_8[0][0]
                                                               leaky_re_lu_22[0][0]
__________________________________________________________________________________________________
conv2d_23 (Conv2D)             (None, None, None, 1 32768    add_9[0][0]
__________________________________________________________________________________________________
batch_normalization_23 (BatchNo (None, None, None, 1 512       conv2d_23[0][0]
__________________________________________________________________________________________________
leaky_re_lu_23 (LeakyReLU)    (None, None, None, 1 0          batch_normalization_23[0][0]
__________________________________________________________________________________________________
conv2d_24 (Conv2D)             (None, None, None, 2 294912    leaky_re_lu_23[0][0]
__________________________________________________________________________________________________
batch_normalization_24 (BatchNo (None, None, None, 2 1024       conv2d_24[0][0]
__________________________________________________________________________________________________
leaky_re_lu_24 (LeakyReLU)    (None, None, None, 2 0          batch_normalization_24[0][0]
__________________________________________________________________________________________________
add_10 (Add)                   (None, None, None, 2 0          add_9[0][0]
                                                               leaky_re_lu_24[0][0]
__________________________________________________________________________________________________
conv2d_25 (Conv2D)             (None, None, None, 1 32768    add_10[0][0]
__________________________________________________________________________________________________
batch_normalization_25 (BatchNo (None, None, None, 1 512       conv2d_25[0][0]
__________________________________________________________________________________________________
leaky_re_lu_25 (LeakyReLU)    (None, None, None, 1 0          batch_normalization_25[0][0]
__________________________________________________________________________________________________
conv2d_26 (Conv2D)             (None, None, None, 2 294912    leaky_re_lu_25[0][0]
__________________________________________________________________________________________________
batch_normalization_26 (BatchNo (None, None, None, 2 1024       conv2d_26[0][0]
__________________________________________________________________________________________________
leaky_re_lu_26 (LeakyReLU)    (None, None, None, 2 0          batch_normalization_26[0][0]
__________________________________________________________________________________________________
add_11 (Add)                   (None, None, None, 2 0          add_10[0][0]
                                                               leaky_re_lu_26[0][0]
__________________________________________________________________________________________________
zero_padding2d_4 (ZeroPadding2D (None, None, None, 2 0          add_11[0][0]
__________________________________________________________________________________________________
conv2d_27 (Conv2D)             (None, None, None, 5 1179648    zero_padding2d_4[0][0]
__________________________________________________________________________________________________
batch_normalization_27 (BatchNo (None, None, None, 5 2048       conv2d_27[0][0]
__________________________________________________________________________________________________
leaky_re_lu_27 (LeakyReLU)    (None, None, None, 5 0          batch_normalization_27[0][0]
__________________________________________________________________________________________________
conv2d_28 (Conv2D)             (None, None, None, 2 131072    leaky_re_lu_27[0][0]
__________________________________________________________________________________________________
batch_normalization_28 (BatchNo (None, None, None, 2 1024       conv2d_28[0][0]
__________________________________________________________________________________________________
leaky_re_lu_28 (LeakyReLU)    (None, None, None, 2 0          batch_normalization_28[0][0]
__________________________________________________________________________________________________
conv2d_29 (Conv2D)             (None, None, None, 5 1179648    leaky_re_lu_28[0][0]
__________________________________________________________________________________________________
batch_normalization_29 (BatchNo (None, None, None, 5 2048       conv2d_29[0][0]
__________________________________________________________________________________________________
leaky_re_lu_29 (LeakyReLU)    (None, None, None, 5 0          batch_normalization_29[0][0]
__________________________________________________________________________________________________
add_12 (Add)                   (None, None, None, 5 0          leaky_re_lu_27[0][0]
                                                               leaky_re_lu_29[0][0]
__________________________________________________________________________________________________
conv2d_30 (Conv2D)             (None, None, None, 2 131072    add_12[0][0]
__________________________________________________________________________________________________
batch_normalization_30 (BatchNo (None, None, None, 2 1024       conv2d_30[0][0]
__________________________________________________________________________________________________
leaky_re_lu_30 (LeakyReLU)    (None, None, None, 2 0          batch_normalization_30[0][0]
__________________________________________________________________________________________________
conv2d_31 (Conv2D)             (None, None, None, 5 1179648    leaky_re_lu_30[0][0]
__________________________________________________________________________________________________
batch_normalization_31 (BatchNo (None, None, None, 5 2048       conv2d_31[0][0]
__________________________________________________________________________________________________
leaky_re_lu_31 (LeakyReLU)    (None, None, None, 5 0          batch_normalization_31[0][0]
__________________________________________________________________________________________________
add_13 (Add)                   (None, None, None, 5 0          add_12[0][0]
                                                               leaky_re_lu_31[0][0]
__________________________________________________________________________________________________
conv2d_32 (Conv2D)             (None, None, None, 2 131072    add_13[0][0]
__________________________________________________________________________________________________
batch_normalization_32 (BatchNo (None, None, None, 2 1024       conv2d_32[0][0]
__________________________________________________________________________________________________
leaky_re_lu_32 (LeakyReLU)    (None, None, None, 2 0          batch_normalization_32[0][0]
__________________________________________________________________________________________________
conv2d_33 (Conv2D)             (None, None, None, 5 1179648    leaky_re_lu_32[0][0]
__________________________________________________________________________________________________
batch_normalization_33 (BatchNo (None, None, None, 5 2048       conv2d_33[0][0]
__________________________________________________________________________________________________
leaky_re_lu_33 (LeakyReLU)    (None, None, None, 5 0          batch_normalization_33[0][0]
__________________________________________________________________________________________________
add_14 (Add)                   (None, None, None, 5 0          add_13[0][0]
                                                               leaky_re_lu_33[0][0]
__________________________________________________________________________________________________
conv2d_34 (Conv2D)             (None, None, None, 2 131072    add_14[0][0]
__________________________________________________________________________________________________
batch_normalization_34 (BatchNo (None, None, None, 2 1024       conv2d_34[0][0]
__________________________________________________________________________________________________
leaky_re_lu_34 (LeakyReLU)    (None, None, None, 2 0          batch_normalization_34[0][0]
__________________________________________________________________________________________________
conv2d_35 (Conv2D)             (None, None, None, 5 1179648    leaky_re_lu_34[0][0]
__________________________________________________________________________________________________
batch_normalization_35 (BatchNo (None, None, None, 5 2048       conv2d_35[0][0]
__________________________________________________________________________________________________
leaky_re_lu_35 (LeakyReLU)    (None, None, None, 5 0          batch_normalization_35[0][0]
__________________________________________________________________________________________________
add_15 (Add)                   (None, None, None, 5 0          add_14[0][0]
                                                               leaky_re_lu_35[0][0]
__________________________________________________________________________________________________
conv2d_36 (Conv2D)             (None, None, None, 2 131072    add_15[0][0]
__________________________________________________________________________________________________
batch_normalization_36 (BatchNo (None, None, None, 2 1024       conv2d_36[0][0]
__________________________________________________________________________________________________
leaky_re_lu_36 (LeakyReLU)    (None, None, None, 2 0          batch_normalization_36[0][0]
__________________________________________________________________________________________________
conv2d_37 (Conv2D)             (None, None, None, 5 1179648    leaky_re_lu_36[0][0]
__________________________________________________________________________________________________
batch_normalization_37 (BatchNo (None, None, None, 5 2048       conv2d_37[0][0]
__________________________________________________________________________________________________
leaky_re_lu_37 (LeakyReLU)    (None, None, None, 5 0          batch_normalization_37[0][0]
__________________________________________________________________________________________________
add_16 (Add)                   (None, None, None, 5 0          add_15[0][0]
                                                               leaky_re_lu_37[0][0]
__________________________________________________________________________________________________
conv2d_38 (Conv2D)             (None, None, None, 2 131072    add_16[0][0]
__________________________________________________________________________________________________
batch_normalization_38 (BatchNo (None, None, None, 2 1024       conv2d_38[0][0]
__________________________________________________________________________________________________
leaky_re_lu_38 (LeakyReLU)    (None, None, None, 2 0          batch_normalization_38[0][0]
__________________________________________________________________________________________________
conv2d_39 (Conv2D)             (None, None, None, 5 1179648    leaky_re_lu_38[0][0]
__________________________________________________________________________________________________
batch_normalization_39 (BatchNo (None, None, None, 5 2048       conv2d_39[0][0]
__________________________________________________________________________________________________
leaky_re_lu_39 (LeakyReLU)    (None, None, None, 5 0          batch_normalization_39[0][0]
__________________________________________________________________________________________________
add_17 (Add)                   (None, None, None, 5 0          add_16[0][0]
                                                               leaky_re_lu_39[0][0]
__________________________________________________________________________________________________
conv2d_40 (Conv2D)             (None, None, None, 2 131072    add_17[0][0]
__________________________________________________________________________________________________
batch_normalization_40 (BatchNo (None, None, None, 2 1024       conv2d_40[0][0]
__________________________________________________________________________________________________
leaky_re_lu_40 (LeakyReLU)    (None, None, None, 2 0          batch_normalization_40[0][0]
__________________________________________________________________________________________________
conv2d_41 (Conv2D)             (None, None, None, 5 1179648    leaky_re_lu_40[0][0]
__________________________________________________________________________________________________
batch_normalization_41 (BatchNo (None, None, None, 5 2048       conv2d_41[0][0]
__________________________________________________________________________________________________
leaky_re_lu_41 (LeakyReLU)    (None, None, None, 5 0          batch_normalization_41[0][0]
__________________________________________________________________________________________________
add_18 (Add)                   (None, None, None, 5 0          add_17[0][0]
                                                               leaky_re_lu_41[0][0]
__________________________________________________________________________________________________
conv2d_42 (Conv2D)             (None, None, None, 2 131072    add_18[0][0]
__________________________________________________________________________________________________
batch_normalization_42 (BatchNo (None, None, None, 2 1024       conv2d_42[0][0]
__________________________________________________________________________________________________
leaky_re_lu_42 (LeakyReLU)    (None, None, None, 2 0          batch_normalization_42[0][0]
__________________________________________________________________________________________________
conv2d_43 (Conv2D)             (None, None, None, 5 1179648    leaky_re_lu_42[0][0]
__________________________________________________________________________________________________
batch_normalization_43 (BatchNo (None, None, None, 5 2048       conv2d_43[0][0]
__________________________________________________________________________________________________
leaky_re_lu_43 (LeakyReLU)    (None, None, None, 5 0          batch_normalization_43[0][0]
__________________________________________________________________________________________________
add_19 (Add)                   (None, None, None, 5 0          add_18[0][0]
                                                               leaky_re_lu_43[0][0]
__________________________________________________________________________________________________
zero_padding2d_5 (ZeroPadding2D (None, None, None, 5 0          add_19[0][0]
__________________________________________________________________________________________________
conv2d_44 (Conv2D)             (None, None, None, 1 4718592    zero_padding2d_5[0][0]
__________________________________________________________________________________________________
batch_normalization_44 (BatchNo (None, None, None, 1 4096       conv2d_44[0][0]
__________________________________________________________________________________________________
leaky_re_lu_44 (LeakyReLU)    (None, None, None, 1 0          batch_normalization_44[0][0]
__________________________________________________________________________________________________
conv2d_45 (Conv2D)             (None, None, None, 5 524288    leaky_re_lu_44[0][0]
__________________________________________________________________________________________________
batch_normalization_45 (BatchNo (None, None, None, 5 2048       conv2d_45[0][0]
__________________________________________________________________________________________________
leaky_re_lu_45 (LeakyReLU)    (None, None, None, 5 0          batch_normalization_45[0][0]
__________________________________________________________________________________________________
conv2d_46 (Conv2D)             (None, None, None, 1 4718592    leaky_re_lu_45[0][0]
__________________________________________________________________________________________________
batch_normalization_46 (BatchNo (None, None, None, 1 4096       conv2d_46[0][0]
__________________________________________________________________________________________________
leaky_re_lu_46 (LeakyReLU)    (None, None, None, 1 0          batch_normalization_46[0][0]
__________________________________________________________________________________________________
add_20 (Add)                   (None, None, None, 1 0          leaky_re_lu_44[0][0]
                                                               leaky_re_lu_46[0][0]
__________________________________________________________________________________________________
conv2d_47 (Conv2D)             (None, None, None, 5 524288    add_20[0][0]
__________________________________________________________________________________________________
batch_normalization_47 (BatchNo (None, None, None, 5 2048       conv2d_47[0][0]
__________________________________________________________________________________________________
leaky_re_lu_47 (LeakyReLU)    (None, None, None, 5 0          batch_normalization_47[0][0]
__________________________________________________________________________________________________
conv2d_48 (Conv2D)             (None, None, None, 1 4718592    leaky_re_lu_47[0][0]
__________________________________________________________________________________________________
batch_normalization_48 (BatchNo (None, None, None, 1 4096       conv2d_48[0][0]
__________________________________________________________________________________________________
leaky_re_lu_48 (LeakyReLU)    (None, None, None, 1 0          batch_normalization_48[0][0]
__________________________________________________________________________________________________
add_21 (Add)                   (None, None, None, 1 0          add_20[0][0]
                                                               leaky_re_lu_48[0][0]
__________________________________________________________________________________________________
conv2d_49 (Conv2D)             (None, None, None, 5 524288    add_21[0][0]
__________________________________________________________________________________________________
batch_normalization_49 (BatchNo (None, None, None, 5 2048       conv2d_49[0][0]
__________________________________________________________________________________________________
leaky_re_lu_49 (LeakyReLU)    (None, None, None, 5 0          batch_normalization_49[0][0]
__________________________________________________________________________________________________
conv2d_50 (Conv2D)             (None, None, None, 1 4718592    leaky_re_lu_49[0][0]
__________________________________________________________________________________________________
batch_normalization_50 (BatchNo (None, None, None, 1 4096       conv2d_50[0][0]
__________________________________________________________________________________________________
leaky_re_lu_50 (LeakyReLU)    (None, None, None, 1 0          batch_normalization_50[0][0]
__________________________________________________________________________________________________
add_22 (Add)                   (None, None, None, 1 0          add_21[0][0]
                                                               leaky_re_lu_50[0][0]
__________________________________________________________________________________________________
conv2d_51 (Conv2D)             (None, None, None, 5 524288    add_22[0][0]
__________________________________________________________________________________________________
batch_normalization_51 (BatchNo (None, None, None, 5 2048       conv2d_51[0][0]
__________________________________________________________________________________________________
leaky_re_lu_51 (LeakyReLU)    (None, None, None, 5 0          batch_normalization_51[0][0]
__________________________________________________________________________________________________
conv2d_52 (Conv2D)             (None, None, None, 1 4718592    leaky_re_lu_51[0][0]
__________________________________________________________________________________________________
batch_normalization_52 (BatchNo (None, None, None, 1 4096       conv2d_52[0][0]
__________________________________________________________________________________________________
leaky_re_lu_52 (LeakyReLU)    (None, None, None, 1 0          batch_normalization_52[0][0]
__________________________________________________________________________________________________
add_23 (Add)                   (None, None, None, 1 0          add_22[0][0]
                                                               leaky_re_lu_52[0][0]
__________________________________________________________________________________________________
conv2d_53 (Conv2D)             (None, None, None, 5 524288    add_23[0][0]
__________________________________________________________________________________________________
batch_normalization_53 (BatchNo (None, None, None, 5 2048       conv2d_53[0][0]
__________________________________________________________________________________________________
leaky_re_lu_53 (LeakyReLU)    (None, None, None, 5 0          batch_normalization_53[0][0]
__________________________________________________________________________________________________
conv2d_54 (Conv2D)             (None, None, None, 1 4718592    leaky_re_lu_53[0][0]
__________________________________________________________________________________________________
batch_normalization_54 (BatchNo (None, None, None, 1 4096       conv2d_54[0][0]
__________________________________________________________________________________________________
leaky_re_lu_54 (LeakyReLU)    (None, None, None, 1 0          batch_normalization_54[0][0]
__________________________________________________________________________________________________
conv2d_55 (Conv2D)             (None, None, None, 5 524288    leaky_re_lu_54[0][0]
__________________________________________________________________________________________________
batch_normalization_55 (BatchNo (None, None, None, 5 2048       conv2d_55[0][0]
__________________________________________________________________________________________________
leaky_re_lu_55 (LeakyReLU)    (None, None, None, 5 0          batch_normalization_55[0][0]
__________________________________________________________________________________________________
conv2d_56 (Conv2D)             (None, None, None, 1 4718592    leaky_re_lu_55[0][0]
__________________________________________________________________________________________________
batch_normalization_56 (BatchNo (None, None, None, 1 4096       conv2d_56[0][0]
__________________________________________________________________________________________________
leaky_re_lu_56 (LeakyReLU)    (None, None, None, 1 0          batch_normalization_56[0][0]
__________________________________________________________________________________________________
conv2d_57 (Conv2D)             (None, None, None, 5 524288    leaky_re_lu_56[0][0]
__________________________________________________________________________________________________
batch_normalization_57 (BatchNo (None, None, None, 5 2048       conv2d_57[0][0]
__________________________________________________________________________________________________
leaky_re_lu_57 (LeakyReLU)    (None, None, None, 5 0          batch_normalization_57[0][0]
__________________________________________________________________________________________________
conv2d_60 (Conv2D)             (None, None, None, 2 131072    leaky_re_lu_57[0][0]
__________________________________________________________________________________________________
batch_normalization_59 (BatchNo (None, None, None, 2 1024       conv2d_60[0][0]
__________________________________________________________________________________________________
leaky_re_lu_59 (LeakyReLU)    (None, None, None, 2 0          batch_normalization_59[0][0]
__________________________________________________________________________________________________
up_sampling2d_1 (UpSampling2D)  (None, None, None, 2 0          leaky_re_lu_59[0][0]
__________________________________________________________________________________________________
concatenate_1 (Concatenate)    (None, None, None, 7 0          up_sampling2d_1[0][0]
                                                               add_19[0][0]
__________________________________________________________________________________________________
conv2d_61 (Conv2D)             (None, None, None, 2 196608    concatenate_1[0][0]
__________________________________________________________________________________________________
batch_normalization_60 (BatchNo (None, None, None, 2 1024       conv2d_61[0][0]
__________________________________________________________________________________________________
leaky_re_lu_60 (LeakyReLU)    (None, None, None, 2 0          batch_normalization_60[0][0]
__________________________________________________________________________________________________
conv2d_62 (Conv2D)             (None, None, None, 5 1179648    leaky_re_lu_60[0][0]
__________________________________________________________________________________________________
batch_normalization_61 (BatchNo (None, None, None, 5 2048       conv2d_62[0][0]
__________________________________________________________________________________________________
leaky_re_lu_61 (LeakyReLU)    (None, None, None, 5 0          batch_normalization_61[0][0]
__________________________________________________________________________________________________
conv2d_63 (Conv2D)             (None, None, None, 2 131072    leaky_re_lu_61[0][0]
__________________________________________________________________________________________________
batch_normalization_62 (BatchNo (None, None, None, 2 1024       conv2d_63[0][0]
__________________________________________________________________________________________________
leaky_re_lu_62 (LeakyReLU)    (None, None, None, 2 0          batch_normalization_62[0][0]
__________________________________________________________________________________________________
conv2d_64 (Conv2D)             (None, None, None, 5 1179648    leaky_re_lu_62[0][0]
__________________________________________________________________________________________________
batch_normalization_63 (BatchNo (None, None, None, 5 2048       conv2d_64[0][0]
__________________________________________________________________________________________________
leaky_re_lu_63 (LeakyReLU)    (None, None, None, 5 0          batch_normalization_63[0][0]
__________________________________________________________________________________________________
conv2d_65 (Conv2D)             (None, None, None, 2 131072    leaky_re_lu_63[0][0]
__________________________________________________________________________________________________
batch_normalization_64 (BatchNo (None, None, None, 2 1024       conv2d_65[0][0]
__________________________________________________________________________________________________
leaky_re_lu_64 (LeakyReLU)    (None, None, None, 2 0          batch_normalization_64[0][0]
__________________________________________________________________________________________________
conv2d_68 (Conv2D)             (None, None, None, 1 32768    leaky_re_lu_64[0][0]
__________________________________________________________________________________________________
batch_normalization_66 (BatchNo (None, None, None, 1 512       conv2d_68[0][0]
__________________________________________________________________________________________________
leaky_re_lu_66 (LeakyReLU)    (None, None, None, 1 0          batch_normalization_66[0][0]
__________________________________________________________________________________________________
up_sampling2d_2 (UpSampling2D)  (None, None, None, 1 0          leaky_re_lu_66[0][0]
__________________________________________________________________________________________________
concatenate_2 (Concatenate)    (None, None, None, 3 0          up_sampling2d_2[0][0]
                                                               add_11[0][0]
__________________________________________________________________________________________________
conv2d_69 (Conv2D)             (None, None, None, 1 49152    concatenate_2[0][0]
__________________________________________________________________________________________________
batch_normalization_67 (BatchNo (None, None, None, 1 512       conv2d_69[0][0]
__________________________________________________________________________________________________
leaky_re_lu_67 (LeakyReLU)    (None, None, None, 1 0          batch_normalization_67[0][0]
__________________________________________________________________________________________________
conv2d_70 (Conv2D)             (None, None, None, 2 294912    leaky_re_lu_67[0][0]
__________________________________________________________________________________________________
batch_normalization_68 (BatchNo (None, None, None, 2 1024       conv2d_70[0][0]
__________________________________________________________________________________________________
leaky_re_lu_68 (LeakyReLU)    (None, None, None, 2 0          batch_normalization_68[0][0]
__________________________________________________________________________________________________
conv2d_71 (Conv2D)             (None, None, None, 1 32768    leaky_re_lu_68[0][0]
__________________________________________________________________________________________________
batch_normalization_69 (BatchNo (None, None, None, 1 512       conv2d_71[0][0]
__________________________________________________________________________________________________
leaky_re_lu_69 (LeakyReLU)    (None, None, None, 1 0          batch_normalization_69[0][0]
__________________________________________________________________________________________________
conv2d_72 (Conv2D)             (None, None, None, 2 294912    leaky_re_lu_69[0][0]
__________________________________________________________________________________________________
batch_normalization_70 (BatchNo (None, None, None, 2 1024       conv2d_72[0][0]
__________________________________________________________________________________________________
leaky_re_lu_70 (LeakyReLU)    (None, None, None, 2 0          batch_normalization_70[0][0]
__________________________________________________________________________________________________
conv2d_73 (Conv2D)             (None, None, None, 1 32768    leaky_re_lu_70[0][0]
__________________________________________________________________________________________________
batch_normalization_71 (BatchNo (None, None, None, 1 512       conv2d_73[0][0]
__________________________________________________________________________________________________
leaky_re_lu_71 (LeakyReLU)    (None, None, None, 1 0          batch_normalization_71[0][0]
__________________________________________________________________________________________________
conv2d_58 (Conv2D)             (None, None, None, 1 4718592    leaky_re_lu_57[0][0]
__________________________________________________________________________________________________
conv2d_66 (Conv2D)             (None, None, None, 5 1179648    leaky_re_lu_64[0][0]
__________________________________________________________________________________________________
conv2d_74 (Conv2D)             (None, None, None, 2 294912    leaky_re_lu_71[0][0]
__________________________________________________________________________________________________
batch_normalization_58 (BatchNo (None, None, None, 1 4096       conv2d_58[0][0]
__________________________________________________________________________________________________
batch_normalization_65 (BatchNo (None, None, None, 5 2048       conv2d_66[0][0]
__________________________________________________________________________________________________
batch_normalization_72 (BatchNo (None, None, None, 2 1024       conv2d_74[0][0]
__________________________________________________________________________________________________
leaky_re_lu_58 (LeakyReLU)    (None, None, None, 1 0          batch_normalization_58[0][0]
__________________________________________________________________________________________________
leaky_re_lu_65 (LeakyReLU)    (None, None, None, 5 0          batch_normalization_65[0][0]
__________________________________________________________________________________________________
leaky_re_lu_72 (LeakyReLU)    (None, None, None, 2 0          batch_normalization_72[0][0]
__________________________________________________________________________________________________
conv2d_59 (Conv2D)             (None, None, None, 2 261375    leaky_re_lu_58[0][0]
__________________________________________________________________________________________________
conv2d_67 (Conv2D)             (None, None, None, 2 130815    leaky_re_lu_65[0][0]
__________________________________________________________________________________________________
conv2d_75 (Conv2D)             (None, None, None, 2 65535    leaky_re_lu_72[0][0]
==================================================================================================
Total params: 62,001,757
Trainable params: 61,949,149
Non-trainable params: 52,608
__________________________________________________________________________________________________
W This network support dynamic shape, but converter do not know the detail,
Please specify input shape,
or manual modify the input shape of model file.
done
--> Building model
W The target_platform is not set in config, using default target platform rk1808.
E Catch exception when building RKNN model!
E Traceback (most recent call last):
E File "/home/kjh/anaconda3/envs/tf1/lib/python3.6/site-packages/tensorflow/python/framework/ops.py", line 1626, in _create_c_op
E    c_op = c_api.TF_FinishOperation(op_desc)
E tensorflow.python.framework.errors_impl.InvalidArgumentError: Negative dimension size caused by subtracting 1 from 0 for 'conv2d_61_49_2/Conv2D' (op: 'Conv2D') with input shapes: [1,0,0,768], [1,1,768,256].
E During handling of the above exception, another exception occurred:
E Traceback (most recent call last):
E File "rknn/api/rknn_base.py", line 578, in rknn.api.rknn_base.RKNNBase.build
E File "rknn/api/rknn_base.py", line 1509, in rknn.api.rknn_base.RKNNBase._quantize2
E File "rknn/base/RKNNlib/RK_nn.py", line 546, in rknn.base.RKNNlib.RK_nn.RKnn.quantize
E File "rknn/base/RKNNlib/app/medusa/quantization.py", line 129, in rknn.base.RKNNlib.app.medusa.quantization.Quantization.run
E File "rknn/base/RKNNlib/app/medusa/quantization.py", line 54, in rknn.base.RKNNlib.app.medusa.quantization.Quantization._run_quantization
E File "rknn/base/RKNNlib/app/medusa/workspace.py", line 150, in rknn.base.RKNNlib.app.medusa.workspace.Workspace.run
E File "rknn/base/RKNNlib/app/medusa/workspace.py", line 120, in rknn.base.RKNNlib.app.medusa.workspace.Workspace._setup_graph
E File "rknn/base/RKNNlib/app/medusa/workspace.py", line 121, in rknn.base.RKNNlib.app.medusa.workspace.Workspace._setup_graph
E File "rknn/base/RKNNlib/RKNNnetbuilder.py", line 310, in rknn.base.RKNNlib.RKNNnetbuilder.RKNNNetBuilder.build
E File "rknn/base/RKNNlib/RKNNnetbuilder.py", line 314, in rknn.base.RKNNlib.RKNNnetbuilder.RKNNNetBuilder.build
E File "rknn/base/RKNNlib/RKNNnetbuilder.py", line 346, in rknn.base.RKNNlib.RKNNnetbuilder.RKNNNetBuilder.build_layer
E File "rknn/base/RKNNlib/RKNNnetbuilder.py", line 346, in rknn.base.RKNNlib.RKNNnetbuilder.RKNNNetBuilder.build_layer
E File "rknn/base/RKNNlib/RKNNnetbuilder.py", line 346, in rknn.base.RKNNlib.RKNNnetbuilder.RKNNNetBuilder.build_layer
E [Previous line repeated 10 more times]
E File "rknn/base/RKNNlib/RKNNnetbuilder.py", line 375, in rknn.base.RKNNlib.RKNNnetbuilder.RKNNNetBuilder.build_layer
E File "rknn/base/RKNNlib/RKNNnetbuilder.py", line 380, in rknn.base.RKNNlib.RKNNnetbuilder.RKNNNetBuilder.build_layer
E File "rknn/base/RKNNlib/layer/RKNNlayer.py", line 290, in rknn.base.RKNNlib.layer.RKNNlayer.RKNNLayer.compute_tensor
E File "rknn/base/RKNNlib/layer/convolution.py", line 122, in rknn.base.RKNNlib.layer.convolution.Convolution.compute_out_tensor
E File "/home/kjh/anaconda3/envs/tf1/lib/python3.6/site-packages/tensorflow/python/ops/gen_nn_ops.py", line 957, in conv2d
E    data_format=data_format, dilations=dilations, name=name)
E File "/home/kjh/anaconda3/envs/tf1/lib/python3.6/site-packages/tensorflow/python/framework/op_def_library.py", line 787, in _apply_op_helper
E    op_def=op_def)
E File "/home/kjh/anaconda3/envs/tf1/lib/python3.6/site-packages/tensorflow/python/util/deprecation.py", line 488, in new_func
E    return func(*args, **kwargs)
E File "/home/kjh/anaconda3/envs/tf1/lib/python3.6/site-packages/tensorflow/python/framework/ops.py", line 3272, in create_op
E    op_def=op_def)
E File "/home/kjh/anaconda3/envs/tf1/lib/python3.6/site-packages/tensorflow/python/framework/ops.py", line 1790, in __init__
E    control_input_ops)
E File "/home/kjh/anaconda3/envs/tf1/lib/python3.6/site-packages/tensorflow/python/framework/ops.py", line 1629, in _create_c_op
E    raise ValueError(str(e))
E ValueError: Negative dimension size caused by subtracting 1 from 0 for 'conv2d_61_49_2/Conv2D' (op: 'Conv2D') with input shapes: [1,0,0,768], [1,1,768,256].
Build model failed!

欢迎光临 Toybrick (https://t.rock-chips.com/)