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

Perform object detection on segmentation area #101

Open
danielzhangau opened this issue Mar 7, 2022 · 1 comment
Open

Perform object detection on segmentation area #101

danielzhangau opened this issue Mar 7, 2022 · 1 comment

Comments

@danielzhangau
Copy link
Contributor

I want to improve the accuracy of object detection by yolop. Is there any way for yolop to directly use the segmented road part for the target detection model after completing the segmentation, or only display the result of the detection target within the segmentation range?

@novoforce
Copy link

hi @danielzhangau from what i understand is YoloP has 3 decoder head which takes the features from the common feature extractor(encoder) and perform 3 different different tasks(detection, segmentation, segmentation). So for your query directly via YoloP is not possible.
Alternatively you can take the output of the 1 decoder segmentation head and perform another detection on top of that or you might have to play around with the architecture of YoloP by adding some residual layer on the segmentation head and perform detection.

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