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.

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

Calculating the number of parameters and FLOPs correctly #178

Open
jameslahm opened this issue Jun 3, 2024 · 9 comments
Open

Calculating the number of parameters and FLOPs correctly #178

jameslahm opened this issue Jun 3, 2024 · 9 comments

Comments

@jameslahm
Copy link
Collaborator

Due to that the one-to-many head in YOLOv10 is not needed during inference, this part of params and FLOPs can be ignored. We provide the script to calculate the number parameters and FLOPs correctly. You could use that like below

python flops.py

Thanks!

@Hy87380510
Copy link

Thanks

@Hy87380510
Copy link

but not have FLOPs
image

@jameslahm
Copy link
Collaborator Author

@Hy87380510 Thanks. The FLOPs are correctly shown in our local environment. Is your code the latest version?

image

@Hy87380510
Copy link

The latest version of the code reports an error
image

@Hy87380510
Copy link

Thank you very much for your help, I have solved the above problem and got the corresponding parameters in the paper.
image

@suyx19
Copy link

suyx19 commented Jun 5, 2024

Thank you very much for your help, I have solved the above problem and got the corresponding parameters in the paper. image

Hi, I have also encountered the same problem. How did you solved it?

@Hy87380510
Copy link

image

@SMDR1412
Copy link

你这操作不是在掩耳盗铃吗?不理解

@JiaDongZou
Copy link

pip install thop

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

5 participants