关于yolov5转换文件

[Zengyf-CVer]

@cmdbug

请问大神，关于#1 这个问题中的第一张图片，这个代码具体是什么文件里的？谢谢

[cmdbug]

yolo.py -> class Detect(nn.Module):

[Zengyf-CVer]

yolo.py -> class Detect(nn.Module):

大神，请问这个yolo.py文件是在什么操作下使用的？
是运行export.py文件导出onnx模型的时候还是运行train.py文件生成pt模型时？

[cmdbug]

只有在导出的时候改一下就可以了，训练时不用改使用默认的就行。

[Zengyf-CVer]

只有在导出的时候改一下就可以了，训练时不用改使用默认的就行。

谢谢大神