search

tusen-ai / simpledet

A Simple and Versatile Framework for Object Detection and Instance Recognition
Apache License 2.0
3.09k stars 487 forks source link

Adding a regression head #323

Closed DecentMakeover closed 4 years ago

DecentMakeover commented 4 years ago

Thanks for sharing your work.

In Faster Rcnn , Along with the classification head and a regression head i would like to add another regression head, how could i go about doing that ,thanks.

zehuichen123 commented 4 years ago

maybe you can refer to this :) https://github.com/TuSimple/simpledet/blob/8da8efe2c2d07e428a81e0b69909b661dfa5bce6/models/msrcnn/builder.py#L65

DecentMakeover commented 4 years ago

Thanks let me check this