跳到正文
掘金· 江润舟·· 23 天前AI 评分38

从零手搓工业级旋转目标检测网络·卷5:YAML 蓝图与 parse_model 动态引擎(二)

万物|炼器:从零手搓工业级旋转目标检测网络·卷5 —— YAML蓝图与parse_model动态引擎(二)

AI 导读

文章讲解 YOLO 类模型如何用 parse_model() 把 YAML 蓝图组装成 PyTorch 网络:通过 globals() 与 getattr() 反射把模块名字符串映射为 Conv、C3k2、SPPF、SimpleHead 等类,避免逐模块写 if-else。

来源:掘金 · juejin.cn