Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

关于yolov5转换文件 #6

Open
Zengyf-CVer opened this issue Jun 28, 2021 · 4 comments
Open

关于yolov5转换文件 #6

Zengyf-CVer opened this issue Jun 28, 2021 · 4 comments

Comments

@Zengyf-CVer
Copy link

Zengyf-CVer commented Jun 28, 2021 •
edited
Loading

@cmdbug

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

image
@cmdbug
Copy link
Owner

cmdbug commented Jun 28, 2021

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

@Zengyf-CVer
Copy link
Author

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

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

@cmdbug
Copy link
Owner

cmdbug commented Jun 28, 2021

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

@Zengyf-CVer
Copy link
Author

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

谢谢大神

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@cmdbug @Zengyf-CVer