lmdb_test.prototxt文件示例内容如下,

name: "LMDB_test"
layer {
  name: "data"
  type: "AnnotatedData"
  top: "data"
  top: "label"
  include {
    phase: TEST
  }
  transform_param {
    scale: 0.00392156862745
    mean_value: 0
    mean_value: 0
    mean_value: 0
    resize_param {
      prob: 1
      resize_mode: WARP
      height: 240
      width: 320
      interp_mode: LINEAR
      height_scale: 240
      width_scale: 320
    }
  }
  data_param {
    source: "./data/test.lmdb"
    batch_size: 8
    backend: LMDB
    threads: 4
    parser_threads: 4
  }
  annotated_data_param {
    label_map_file: "./data/label_map_file.prototxt"
  }
}

layer {
  name: "output"
  type: "DataOutput"
  bottom: "data"
  bottom: "label"
  data_output_param {
    data_type: DETECTION
  }
}

使用Netron工具显示如下,

参考:

Netscope

模型结构可视化神器——Netron(支持tf, caffe, keras,mxnet等多种框架)_木盏的博客-CSDN博客_netron

轻量好用的神经网络模型可视化工具netron_Mingyong_Zhuang的博客-CSDN博客_神经网络模型可视化

Have Fun

Logo

永洪科技,致力于打造全球领先的数据技术厂商,具备从数据应用方案咨询、BI、AIGC智能分析、数字孪生、数据资产、数据治理、数据实施的端到端大数据价值服务能力。

更多推荐