search

cmdbug / TNN_Demo

🍉 移动端TNN部署学习笔记，支持Android与iOS。
GNU General Public License v3.0
69 stars 18 forks source link

关于yolov5转换文件 #6

Open Zengyf-CVer opened 3 years ago

Zengyf-CVer commented 3 years ago

@cmdbug

请问大神，关于https://github.com/cmdbug/TNN_Demo/issues/1 这个问题中的第一张图片，这个代码具体是什么文件里的？谢谢

image

cmdbug commented 3 years ago

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

Zengyf-CVer commented 3 years ago

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

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

cmdbug commented 3 years ago

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

Zengyf-CVer commented 3 years ago

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

谢谢大神