当前位置:   article > 正文

海思SD3403,SS928/926,hi3519dv500,hi3516dv500移植yolov7,yolov8(9)_sd3403 npu算力

sd3403 npu算力

 上一节yolov8的训练已经完成了,现在要开始做模型的转换了,这里和yolov7方式相似,但是有一些差异,尤其是yolov7的不带NMS部分的输出顺序和yolov8的输出顺序与格式是有差异的。

   首先还是要自己手动加入rpn_op,这里包含了filter,sort,nms部分。

我们一个一个看,首先filter.py里 要先增加个Filter的自定义层。 

  1. op_source = """
  2. #include <torch/script.h>
  3. torch::Tensor Filter(torch::Tensor in, int64_t topK, double m_lowScoreThresh)
  4. {
  5. return in;
  6. }
  7. static auto registry =
  8. torch::RegisterOperators("custom_ops::Filter", &Filter);
  9. """
  10. torch.utils.cpp_extension.load_inline(
  11. name='Filter',
  12. cpp_sources=op_source,
  13. is_python_module=False,
  14. verbose=True,
  15. )
  16. @parse_args("v", "i", "f")
  17. def symbolic_filter
声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/Cpp五条/article/detail/724062
推荐阅读
相关标签
  

闽ICP备14008679号