更新时间:2022-02-22 GMT+08:00
为什么在ModelArts中验证精度很高的模型,在HiLens Studio中测试效果却很差?
如果是使用ModelArts的预置算法或者AI Gallery上的算法,出现精度差异问题可能有以下几个原因。
- 预处理问题:比如模型训练时用的是RGB格式,在HiLens推理时用了BGR;模型输入是等比例缩放加padding(尤其是对于416*416这种方形输入),而HiLens里面是直接缩放。
模型的正确预处理步骤可以在ModelArts训练输出目录中的“customize_service.py”中找到。
- 参数问题:比如yolo3模型在解码时依赖anchors,这些anchors必须和训练时的anchors保持一致,同样的这些参数可以在ModelArts训练输出目录中的“customize_service.py”中找到。
- 类别顺序问题:当输出结果时类别顺序一定要与模板中的类别顺序一致。如果自己的模型输出顺序不一样,做一个简单的映射即可,比如“speed_limited”类,在模板中是第 0 个,在你的模型中是第 3 个,那么当你推理得到类别 3 的检测框时,输出时将类别改为 0 。你的模型类别顺序可以在ModelArts训练输出目录中的“index”文件中找到。
父主题: 算法类问题