Toybrick

caffe转rknn出错

虚无灵幻

注册会员

积分
71
楼主
发表于 2022-2-18 09:35:06    查看: 3054|回复: 1 | [复制链接]    打印 | 只看该作者
root@f3231724c48f:/test/rknn_openpose# python3 rknn_transfer.py
--> config model
done
--> Loading model
W Unknown caffe layer loss_stage1_L1, op is euclideanloss, please REMOVE it from prototxt if possible.
W Unknown caffe layer loss_stage1_L2, op is euclideanloss, please REMOVE it from prototxt if possible.
W Unknown caffe layer loss_stage2_L1, op is euclideanloss, please REMOVE it from prototxt if possible.
W Unknown caffe layer loss_stage2_L2, op is euclideanloss, please REMOVE it from prototxt if possible.
W Unknown caffe layer loss_stage3_L1, op is euclideanloss, please REMOVE it from prototxt if possible.
W Unknown caffe layer loss_stage3_L2, op is euclideanloss, please REMOVE it from prototxt if possible.
W Unknown caffe layer loss_stage4_L1, op is euclideanloss, please REMOVE it from prototxt if possible.
W Unknown caffe layer loss_stage4_L2, op is euclideanloss, please REMOVE it from prototxt if possible.
W Unknown caffe layer loss_stage5_L1, op is euclideanloss, please REMOVE it from prototxt if possible.
W Unknown caffe layer loss_stage5_L2, op is euclideanloss, please REMOVE it from prototxt if possible.
W Unknown caffe layer loss_stage6_L1, op is euclideanloss, please REMOVE it from prototxt if possible.
W Unknown caffe layer loss_stage6_L2, op is euclideanloss, please REMOVE it from prototxt if possible.
E Catch exception when loading caffe model: ./pose_train_test.prototxt!
E Traceback (most recent call last):
E   File "rknn/api/rknn_base.py", line 261, in rknn.api.rknn_base.RKNNBase.load_caffe
E   File "rknn/base/RKNNlib/RK_nn.py", line 74, in rknn.base.RKNNlib.RK_nn.RKnn.load_caffe
E   File "rknn/base/RKNNlib/RKNNnetbuilder.py", line 400, in rknn.base.RKNNlib.RKNNnetbuilder.RKNNNetBuilder.build
E   File "rknn/base/RKNNlib/RKNNnetbuilder.py", line 426, in rknn.base.RKNNlib.RKNNnetbuilder.RKNNNetBuilder.build_layer
E   File "rknn/base/RKNNlib/RKNNnetbuilder.py", line 426, in rknn.base.RKNNlib.RKNNnetbuilder.RKNNNetBuilder.build_layer
E   File "rknn/base/RKNNlib/RKNNnetbuilder.py", line 438, in rknn.base.RKNNlib.RKNNnetbuilder.RKNNNetBuilder.build_layer
E   File "rknn/base/RKNNlib/layer/RKNNlayer.py", line 265, in rknn.base.RKNNlib.layer.RKNNlayer.RKNNLayer.compute_shape
E   File "rknn/base/RKNNlib/layer/eltwise.py", line 42, in rknn.base.RKNNlib.layer.eltwise.EltWise.compute_out_shape
E AttributeError: 'NoneType' object has no attribute 'shape'
Load model failed!
root@f3231724c48f:/test/rknn_openpose# python3 rknn_transfer.py
--> config model
done
--> Loading model
W Unknown caffe layer loss_stage1_L1, op is euclideanloss, please REMOVE it from prototxt if possible.
W Unknown caffe layer loss_stage1_L2, op is euclideanloss, please REMOVE it from prototxt if possible.
W Unknown caffe layer loss_stage2_L1, op is euclideanloss, please REMOVE it from prototxt if possible.
W Unknown caffe layer loss_stage2_L2, op is euclideanloss, please REMOVE it from prototxt if possible.
W Unknown caffe layer loss_stage3_L1, op is euclideanloss, please REMOVE it from prototxt if possible.
W Unknown caffe layer loss_stage3_L2, op is euclideanloss, please REMOVE it from prototxt if possible.
W Unknown caffe layer loss_stage4_L1, op is euclideanloss, please REMOVE it from prototxt if possible.
W Unknown caffe layer loss_stage4_L2, op is euclideanloss, please REMOVE it from prototxt if possible.
W Unknown caffe layer loss_stage5_L1, op is euclideanloss, please REMOVE it from prototxt if possible.
W Unknown caffe layer loss_stage5_L2, op is euclideanloss, please REMOVE it from prototxt if possible.
W Unknown caffe layer loss_stage6_L1, op is euclideanloss, please REMOVE it from prototxt if possible.
W Unknown caffe layer loss_stage6_L2, op is euclideanloss, please REMOVE it from prototxt if possible.
E Catch exception when loading caffe model: ./pose_train_test.prototxt!
E Traceback (most recent call last):
E   File "rknn/api/rknn_base.py", line 261, in rknn.api.rknn_base.RKNNBase.load_caffe
E   File "rknn/base/RKNNlib/RK_nn.py", line 74, in rknn.base.RKNNlib.RK_nn.RKnn.load_caffe
E   File "rknn/base/RKNNlib/RKNNnetbuilder.py", line 400, in rknn.base.RKNNlib.RKNNnetbuilder.RKNNNetBuilder.build
E   File "rknn/base/RKNNlib/RKNNnetbuilder.py", line 426, in rknn.base.RKNNlib.RKNNnetbuilder.RKNNNetBuilder.build_layer
E   File "rknn/base/RKNNlib/RKNNnetbuilder.py", line 426, in rknn.base.RKNNlib.RKNNnetbuilder.RKNNNetBuilder.build_layer
E   File "rknn/base/RKNNlib/RKNNnetbuilder.py", line 438, in rknn.base.RKNNlib.RKNNnetbuilder.RKNNNetBuilder.build_layer
E   File "rknn/base/RKNNlib/layer/RKNNlayer.py", line 265, in rknn.base.RKNNlib.layer.RKNNlayer.RKNNLayer.compute_shape
E   File "rknn/base/RKNNlib/layer/eltwise.py", line 42, in rknn.base.RKNNlib.layer.eltwise.EltWise.compute_out_shape
E AttributeError: 'NoneType' object has no attribute 'shape'
Load model failed!
回复

使用道具 举报

虚无灵幻

注册会员

积分
71
沙发
 楼主| 发表于 2022-2-18 09:47:12 | 只看该作者
layer {
  name: "data"
  type: "CPMData"
  top: "data"
  top: "label"
  data_param {
    source: "/home/zhecao/COCO_kpt/lmdb_trainVal"
    batch_size: 10
    backend: LMDB
  }
  cpm_transform_param {
    stride: 8
    max_rotate_degree: 40
    visualize: false
    crop_size_x: 368
    crop_size_y: 368
    scale_prob: 1
    scale_min: 0.5
    scale_max: 1.1
    target_dist: 0.6
    center_perterb_max: 40
    do_clahe: false
    num_parts: 56
    np_in_lmdb: 17
  }
}
layer {
  name: "vec_weight"
  type: "Slice"
  bottom: "label"
  top: "vec_weight"
  top: "heat_weight"
  top: "vec_temp"
  top: "heat_temp"
  slice_param {
    slice_point: 38
    slice_point: 57
    slice_point: 95
    axis: 1
  }
}
layer {
  name: "label_vec"
  type: "Eltwise"
  bottom: "vec_weight"
  bottom: "vec_temp"
  top: "label_vec"
  eltwise_param {
    operation: PROD
  }
}
layer {
  name: "label_heat"
  type: "Eltwise"
  bottom: "heat_weight"
  bottom: "heat_temp"
  top: "label_heat"
  eltwise_param {
    operation: PROD
  }
}
layer {
  name: "image"
  type: "Slice"
  bottom: "data"
  top: "image"
  top: "center_map"
  slice_param {
    slice_point: 3
    axis: 1
  }
}
layer {
  name: "silence2"
  type: "Silence"
  bottom: "center_map"
}
layer {
  name: "conv1_1"
  type: "Convolution"
  bottom: "image"
  top: "conv1_1"
  param {
    lr_mult: 1.0
    decay_mult: 1
  }
原来的prototxt怎么改
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

产品中心 购买渠道 开源社区 Wiki教程 资料下载 关于Toybrick


快速回复 返回顶部 返回列表