关于yolov5转换文件

[Zengyf-CVer]

@cmdbug

请问大神，关于https://github.com/cmdbug/TNN_Demo/issues/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]

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

谢谢大神