结庐东方 · 2021年07月19日

【周易AIPU 仿真】R329开发板申请——模型仿真测试

  1. 原始模型文件
    ResNet V1 50
    http://download.tensorflow.or...
  2. 矫正集
    dataset.npy和label.npy矫正集文件
    链接: https://pan.baidu.com/s/1n-sv... 提取码: vadh
  3. cfg 文件
    resnet_50_run.cfg
    链接: https://pan.baidu.com/s/1kNZ5... 提取码: 2umb

    [Common]
    mode=run
    
    [Parser]
    model_name = resnet_50
    detection_postprocess = 
    model_domain = image_classification
    output = resnet_v1_50/predictions/Reshape_1
    input_model = ./resnet_v1_50_frozen.pb
    input = input
    input_shape = [1,224,224,3]
    output_dir = ./
    
    [AutoQuantizationTool]
    model_name = resnet_50
    quantize_method = SYMMETRIC
    ops_per_channel = DepthwiseConv
    calibration_data = ./dataset.npy
    calibration_label = ./label.npy
    preprocess_mode = normalize
    quant_precision=int8
    reverse_rgb = False
    label_id_offset = 0
    
    [GBuilder]
    inputs=./input.bin
    simulator=aipu_simulator_z1
    outputs=output_resnet_50.bin
    profile= True
    target=Z1_0701
  4. simulator 执行结果
    aipubuild resnet_50_run​.cfg
    运行之后得到 output_resnet_50.bin,使用 quant_predict.py 解析后:

    predict first 5 label:
     index  230, prob  11, name: Old English sheepdog, bobtail
     index  232, prob   2, name: collie
     index  224, prob   1, name: schipperke
     index  999, prob   0, name: ear, spike, capitulum
     index  329, prob   0, name: sea urchin
    true first 5 label:
     index  230, prob 109, name: Old English sheepdog, bobtail
     index  231, prob  96, name: Shetland sheepdog, Shetland sheep dog, Shetland
     index  232, prob  57, name: collie
     index  226, prob  54, name: malinois
     index  263, prob  53, name: Brabancon griffon
    Detect picture save to result.jpeg

    测试图片:
    result.jpeg
    5.模型实机测试
    待审核通过收到实机后测试

推荐阅读
关注数
7434
内容数
92
人工智能边缘计算软硬件解决方案,提供高性能、低成本、低功耗、易使用的硬件选型方案.
目录
极术微信服务号
关注极术微信号
实时接收点赞提醒和评论通知
安谋科技学堂公众号
关注安谋科技学堂
实时获取安谋科技及 Arm 教学资源
安谋科技招聘公众号
关注安谋科技招聘
实时获取安谋科技中国职位信息