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

TPH #4

Closed
sandy-0326 opened this issue Nov 25, 2021 · 2 comments
Closed

TPH #4

sandy-0326 opened this issue Nov 25, 2021 · 2 comments

Comments

@sandy-0326
Copy link

sandy-0326 commented Nov 25, 2021

I want to know where the TPH module is?
And I want to know whether the network structure described in the file "yolov5l-xs-tr-cbam-spp-bifpn.yaml" is consistent with the network structure described in the paper.

@cv516Buaa
Copy link
Owner

Transformer prediction head is the combination of transformer block with yolo pretiction head,
image
C3TR block with the prediction head together form TPH.
"yolov5l-xs-tr-cbam-spp-bifpn.yaml" is not consistent with network structure described in the paper, it only use one tph module with some extra layer for the balance of speed and precision,but the structure in paper use 4 tph module.

@sandy-0326
Copy link
Author

Transformer prediction head is the combination of transformer block with yolo pretiction head, image C3TR block with the prediction head together form TPH. "yolov5l-xs-tr-cbam-spp-bifpn.yaml" is not consistent with network structure described in the paper, it only use one tph module with some extra layer for the balance of speed and precision,but the structure in paper use 4 tph module.
Thank you! I get it.

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