TI中文支持网
TI专业的中文技术问题搜集分享网站

TDA4VM: 基于Yolov5模型的后处理问题。

Part Number:TDA4VM

我们使用自己基于YoloV5训练的模型,该模型的输出特征维度与目前TIDL库中默认支持的维度不同,在使用TIDL的tidl_model_import.out工具进行转换时,需要修改TIDL_findValidLocation_cn()函数中的numInCodeElements变量的值才能得到正确的结果,但是在使用转换后的模型在TDA4VM开发板上运行时,发现程序会调用countM = pKerPrivArgsL->execute((void *)pKerPrivArgsL, NULL, NULL, NULL, NULL)函数,我们初步定位该步骤会执行pFxnTIDL_odFindValidLocAndScore_ixX_oxX_exec函数指针指向的函数,我们目前没有找到函数的定义,请问是否能提供该函数的具体实现?若无具体实现,请问与pc端的函数实现(即TIDL_findValidLocation_cn()函数)有何不同?望得到帮助。

Katherine Wang:

已为您联系TI资深工程师,一旦得到回复会立刻回复给您

,

Katherine Wang:

当您在 PC 上运行推理时,它默认以调用 _cn() 函数的引用模式运行。此流程对您的模型有用吗?

您可以在推理 cfg 文件中设置 flowCtrl=1 以在硬件上运行参考流,否则将运行 DSP 优化代码。我会建议确保您首先在 PC 上获得预期的功能。

关于您的具体问题——您可以在整个项目中搜索函数 TIDL_odFindValidLocAndScore_exec_c7x(),您应该能够在文件 <tidl_dir>/ti_dl/algo/src/tidsp/c7x/tidl_odFindValidLocAndScore_ixX_c7x.c 中找到它。

赞(0)
未经允许不得转载:TI中文支持网 » TDA4VM: 基于Yolov5模型的后处理问题。
分享到: 更多 (0)