NNCASE模型转换后效果不正常



  • Hi All,
    我用了论坛帖子中提到的https://github.com/zhen8838/k210-face-detection仓库,训练好了pb模型,在export-pb步骤,因为kendryte-model-compiler仓库现在已经被删除,无法下载,改用nncase转换,转换时命令如下:

    toco --graph_def_file=workspace/Freeze_save.pb --input_format=TENSORFLOW_GRAPHDEF --output_format=TFLITE --output_file=workspace/Freeze_save.tflite --inference_type=FLOAT --input_type=FLOAT --input_arrays=Input_image --output_arrays=Yolo/Final/conv2d/BiasAdd --input_shapes=1,240,320,3
    
    ncc -i tflite -o k210model --dataset ../../../../k210-face-detection/FDDB/ ./workspace/Freeze_save.tflite ./workspace/Freeze_save.kmodel
    

    然后跟把standalone-demo中,face_detection里面的kfpkg文件重命名成zip解压,替换中间的kmodel文件,修改json文件如下:

    {
      "version": "0.1.0",
      "files": [
        {
          "address": 0,
          "bin": "face_detect.bin",
          "sha256Prefix": true
        },
        {
          "address": 0x00A00000,
          "bin": "Freeze_save.kmodel",
          "sha256Prefix": false
        }
      ]
    }
    

    然后重新打包成zip文件,重命名成kfpkg文件,用K-Flash.exe下载到KD233开发板,检测出现异常:
    0_1566981311103_微信图片编辑_20190828163429.jpg
    摄像头采集显示刚开始就卡住了,偶尔可以流畅播放,但检测框一直固定不变,位置明显不对。
    麻烦帮忙看下,操作过程哪里有问题吗?谢谢~


  • Staff |  Mod

    Maybe, the model you used here will use different anchors or other parameters from kendryte-standalone-demos/face_detect. Seeking help from the model's author is recommended.



  • 另外请教下,

    ncc -i tflite -o k210model --dataset ../../../../k210-face-detection/FDDB/ ./workspace/Freeze_save.tflite ./workspace/Freeze_save.kmodel 
    

    这里的FDDB仓库要额外处理吗?怎么处理?