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

anchor define #161

Open
alicera opened this issue Oct 30, 2022 · 3 comments
Open

anchor define #161

alicera opened this issue Oct 30, 2022 · 3 comments

Comments

@alicera
Copy link

alicera commented Oct 30, 2022

yolov7
I cant find the yolov7 anchor
How do you compute the anchor?

https://github.com/WongKinYiu/yolov7/blob/main/cfg/training/yolov7.yaml

@linghu8812
Copy link
Owner

yolov7 I cant find the yolov7 anchor How do you compute the anchor?

https://github.com/WongKinYiu/yolov7/blob/main/cfg/training/yolov7.yaml

it has been export in ONNX model and do not need to write in config files.

image

@alicera
Copy link
Author

alicera commented Oct 31, 2022

Could you tell me where is the possible place that anchor

I try to read the onnx

python export.py --weights ast.pt --simplify

Inputs:  ['images']
Outputs:  ['output', '516', '530'] 

input name images
input shape [1, 3, 640, 640]
input type tensor(float) 

output name: output
output shape: [1, 3, 80, 80, 85]
output type: tensor(float) 

output name: 516
output shape: [1, 3, 40, 40, 85]
output type: tensor(float) 

output name: 530
output shape: [1, 3, 20, 20, 85]
output type: tensor(float) 

@linghu8812
Copy link
Owner

Could you tell me where is the possible place that anchor

I try to read the onnx

python export.py --weights ast.pt --simplify

Inputs:  ['images']
Outputs:  ['output', '516', '530'] 

input name images
input shape [1, 3, 640, 640]
input type tensor(float) 

output name: output
output shape: [1, 3, 80, 80, 85]
output type: tensor(float) 

output name: 516
output shape: [1, 3, 40, 40, 85]
output type: tensor(float) 

output name: 530
output shape: [1, 3, 20, 20, 85]
output type: tensor(float) 

view the output ONNX file with netron
https://netron.app/

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

2 participants