YOLOP icon indicating copy to clipboard operation
YOLOP copied to clipboard

Published 20 hours ago • verified publisher icon hustvl

Possible to add a regression head after the detection head?

Open ithmz opened this issue 4 years ago • 4 comments

As title

ithmz avatar Sep 17 '21 05:09 ithmz

Thanks for your attention to our project! But I'm sorry I don't quite understand what you mean by regression head? Can you tell me more about it?

Riser6 avatar Sep 18 '21 01:09 Riser6

Hi @Riser6 I mean, for example, your detection head can detect pedestrians and cars. Then I want to add another head after the detection head to find the human pose if the class is pedestrian. Is this possible?

ithmz avatar Sep 22 '21 08:09 ithmz

Sure, as long as these downstream tasks are differentiable and can be optimized

Riser6 avatar Sep 22 '21 13:09 Riser6

@Riser6, thanks for your reply Could you give any hints on how to add another head?

ithmz avatar Sep 23 '21 07:09 ithmz