在Yolov8中model.export后self.export=false问题(记录)

本文主要是介绍在Yolov8中model.export后self.export=false问题(记录)，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

遇到一个问题是自己创建了新的Detect检测头，但是在导出模型时，想要修改输出格式，在yolo中可以通过if self.export:来修改网络的返回值格式

当使用model.export()导出时，理论上会自动将export设置为True

但是在实际中发现export=false,于是通过调试发现在ultralytics/engine/exporter.py中

for m in model.modules():if isinstance(m, (Detect, RTDETRDecoder)):  # includes all Detect subclasses like Segment, Pose, OBBm.dynamic = self.args.dynamicm.export = Truem.format = self.args.formatelif isinstance(m, C2f) and not any((saved_model, pb, tflite, edgetpu, tfjs)):# EdgeTPU does not support FlexSplitV while split provides cleaner ONNX graphm.forward = m.forward_split

需要对自己的检测头进行实例，加入到代码中，如下

if isinstance(m, (Detect, RTDETRDecoder，MyDetect)):

再次执行model.export() self.export则为True

但是这个方法并不是最规范的，设置自己的类继承Detect会更好MyDetect（Detect）

参考地址：github

这篇关于在Yolov8中model.export后self.export=false问题(记录)的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！

---