专栏算法工具链hat模块注册问题咨询

hat模块注册问题咨询

已解决
默认383362025-04-28
59
6

我想用工具链提供的config方法和hat工具包用QAT量化yolov5，我在hat/models/structures/detectors下模仿yolov3新建yolov5类，且在init中import了。在config中使用时显示YOLOV5 has not registered in any of registry ['HAT_OBJECT_REGISTRY'] and is not a class, which is not allowed。为了排除我yolov5类写法上的错误，我将yolov3的类复制了一遍，仅把名字换成yolov5，但还是无法注册。我在hat/models下的其他文件夹注册的类，如loss等均可以，只有detectors下这处不行，请问是什么原因？

算法工具链
征程5
评论4
0/1000
  • Huanghui
    Lv.5
    你好，你参考下j6里面自定义模型注册。
    2025-04-28
    0
    1
    • Huanghui回复Huanghui:

      这个是添加自定义模型

      2025-04-28
      0
  • Huanghui
    Lv.5

    这个是添加config文件

    2025-04-28
    0
    0
  • Huanghui
    Lv.5

    提个建议哈，HAT是我们考虑参考算法的统一性构建的，如果你自己想用QAT方式量化 yolov5 ，直接用户plugin方式就OK了。

    2025-04-29
    0
    0
  • Huanghui
    Lv.5

    另外，不要在 hat/models/structures/detectors 下面注册是你自己的类，如果后面升级，你注册的东西会被移除调的，除非你后面不升级，但是这是不现实的！最好的方法就是直接使用plugin方式使用qat，如果就是想用HAT，那就使用文档中说的 自定义模型注册的方法

    2025-04-29
    0
    1
    • 默认38336回复Huanghui:

      您说的plugin是按官方文档里QAT下mobilenet示例的方式，使用 horizon_plugin_pytorch吗

      2025-04-29
      0