[参考译文] TDA4VM：Yolov5/8 纳米模型训练/推理

我想对模型进行微调、而不是从头开始训练

尊敬的 Vendanth：

您使用的是哪个 TIDL 版本？  您无法在电路板上进行训练或编译。  您可以在主机上编译和训练、并在电路板上运行推理。  请包括模型（首选）或您从哪里获得。  模型中可能存在不受支持的运算符。

此致、

Chris

您好、我们使用的是 tidl 版本 11_00_06_00、我知道我们无法在电路板上进行训练和编译。

• 我是在主机上执行此操作的。
• 是否有任何关于 yolov5n 和 yolov8n 流程的文档
• 我正在寻找微调从超声学下载的模型,我们知道我们必须做模型手术,看看如何在预训练的模型或模型训练我们使用 Pytorch

尊敬的 Vedanth：

您可能需要使用模型优化将模型转换为 lite 格式。  我包含了一个用于此目的的脚本、但在执行此操作之前  、您可能需要克隆 edgeai-tensorlab 并安装一些模块。

Git 克隆 github.com/.../edgeai-tensorlab.git
CD edgeai-tensorlab/
CD edgeai-modeloptimization/
PIP 安装。

PIP 安装 Torchvision

PIP 安装 tqdm

PIP 安装 netron

您需要将模型更新为适用于 TIDL 的格式的主要调用是：

edgeai_torchmodelopt.xmodelopt.surv1.convert_to_lite_model ()

#手术
如果 args.model_surgery ==xmodelopt.surgy.SyrgeryVersion.surgation_V1：
  model = xmodelopt.surv1.convert_to_lite_model (model)
  name_suffix =“_lite.onnx"</s>“
elif args.model_surgery ==xmodelopt.surgy.SyrgeryVersion.surgation_fx：
  model = xmodelopt.surv2.convert_to_lite_fx (model)
  name_suffix =“_lite.onnx"</s>“
否则：
  提起 RuntimeError (“不支持的手术类型“）
#

然后保存您的模型并重试导入。

这些是 TIDL 支持的 yolo5 模型。

EdgeAI-YOLOv5 模型

• 培训源代码和模型
• 原始源代码：我们对该存储库进行了更改以创建 edgeai-yolov5 fork。

YOLOv5 模型具有很高的精度、是低复杂性推理的理想选择。 我们优化了 YOLOv5 模型、使其适合 TIDL。 请查看上面的链接以了解更多信息、尤其是在“由 TI 训练的模型“一节下。 还在此处发布经过预先训练的模型检查点和预编译的模型工件。

有一些支持 yolov8 的变体,但由于许可限制,我们的网站上不可用。  

e2e.ti.com/.../model_5F00_surgery_5F00_examples.py

此致、

嗨、我尝试了一些建议的事情、这些都是我有疑问的几件事

• TIDL 是否也支持 yolov5n 型号、还是仅支持 s 以后的型号？
• 我尝试在一个经过预先训练的优洛沃 5 模型上执行模型手术,从超透析学,在你提供的脚本的帮助下。 虽然我没有在 edgeai-tidl-tools 生成的输出映像中看到任何边界框、但编译正在进行中。
• 我已经使用 edgeai-yolov5 repo 来微调训练 yolov5s 模型,它似乎工作正常,有类似的可能性为 yolov5n 吗? 因为 yolov5n 有一个不同的主干,并在 v6.0 中引入  
• TIDL 是否完全支持 yolov5n？

尊敬的 Vendanth：

• TIDL 是否也支持 yolov5n 型号、还是仅支持 s 以后的型号？
  • TIDL 通过模型优化器和外科手术支持 S 型号、您可以使 yolo5n 正常工作
• 我尝试在一个经过预先训练的优洛沃 5 模型上执行模型手术,从超透析学,在你提供的脚本的帮助下。 虽然我没有在 edgeai-tidl-tools 生成的输出映像中看到任何边界框、但编译正在进行中
  • 很高兴听到这个帮助。  为边界框和对象 ID 查找与模型关联的 prototex 文件
• 我已经使用 edgeai-yolov5 repo 来微调训练 yolov5s 模型,它似乎工作正常,有类似的可能性为 yolov5n 吗? 因为 yolov5n 有一个不同的主干,并在 v6.0 中引入  
  • 培训应同样进行
• TIDL 是否完全支持 yolov5n？
  • 请参阅第一个要点

我建议从 Model Zoo 中已有的模型开始、因为它们 开箱即可获得支持。  用一个随机的模型从野外会花你更长的时间。

此致、

Chris

我能够从头开始训练 yolov5n 如果我改变 yolov5n.yaml 匹配骨干从 v5.0,但我想微调它,我无法从 edgeai-yolov5 存储库,因为 yolov5n 是在 v6.0 的超分析与不同的骨干层和不同的层,例如 SPPF , 这在旧版本中不存在、因此我收到一个错误、指出当我尝试从 edgeai-yolov5 对其进行微调时、这些层不存在、是否有预先训练的 yolov5n 模型可用？