专栏算法工具链图像分割mobileunet模型训练方案的说明

图像分割mobileunet模型训练方案的说明

已解决
zhengsiyu2023-04-14
40
2

有没有关于图像分割mobileunet模型训练方案的说明，官方提供的开发手册中几乎什么都没有说，参考网上的模型训练过程训练出来的模型在模型转换完成之后推理得到的结果与预期输出完全不同，有没有大佬指点一二。

例如：

原图：

预期效果：

实际板端推理结果：

算法工具链
评论2
0/1000
  • 颜值即正义
    Lv.2

    您好，示例模型的训练策略已经在config文件中配置好了，一般情况下不需要进行过多修改。首先，您这边可以使用predict.py脚本来验证示例的浮点、qat、量化模型精度，确定一下是否是因为量化损失过大，如果量化损失过大，可以使用lr和epoch调整、量化参数初始化、数据增强等方法来重新训练；然后，可以在PC端运行python3 tools/infer.py -c configs/segmentation/unet.py --input-images $images$ --dataset cityscapes --input-size $input size$ --input-format yuv --is-plot来查看一下可视化效果，并与板端结果对比一下。

    2023-04-17
    0
    0
  • 颜值即正义
    Lv.2
    2023-04-24
    0
    0