上一节yolov8的训练已经完成了，现在要开始做模型的转换了，这里和yolov7方式相似，但是有一些差异，尤其是yolov7的不带NMS部分的输出顺序和yolov8的输出顺序与格式是有差异的。
首先还是要自己手动加入rpn_op,这里包含了filter,sort,nms部分。
文章来源：https://www.toymoban.com/news/detail-796740.html
我们一个一个看，首先filter.py里 要先增加个Filter的自定义层。 文章来源地址https://www.toymoban.com/news/detail-796740.html
op_source = """
#include <torch/script.h>
torch::Tensor Filter(torch::Tensor in, int64_t topK, double m_lowScoreThresh)
{
return in;
}
static auto registry =
torch::RegisterOperators("custom_ops::Filter", &Filter);
"""
torch.utils.cpp_extension.load_inline(
name='Filter',
cpp_sources=op_source,
is_python_module=False,
verbose=True,
)
@parse_args("v", "i", "f")
def symbolic_filter到了这里，关于海思SD3403,SS928/926,hi3519dv500,hi3516dv500移植yolov7,yolov8（9）的文章就介绍完了。如果您还想了解更多内容，请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章，希望大家以后多多支持TOY模板网！
