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

Bounding Box Regression #101

Open
SamihaSara opened this issue Mar 31, 2021 · 1 comment
Open

Bounding Box Regression #101

SamihaSara opened this issue Mar 31, 2021 · 1 comment
Labels
question Further information is requested

Comments

@SamihaSara
Copy link

SamihaSara commented Mar 31, 2021

Does this version of FasterRCNN do bounding box regression? Could you please point where I can find the related codes? If it is not implemented already, could you please consider adding this feature? Thanks.

@SamihaSara SamihaSara added the enhancement New feature or request label Mar 31, 2021
@oke-aditya oke-aditya added question Further information is requested and removed enhancement New feature or request labels Mar 31, 2021
@oke-aditya
Copy link
Owner

oke-aditya commented Mar 31, 2021

Hello @SamihaSara ! Thank you for brining this question!

The FasterRCNN used here is from torchvision.
You can find FasterRCNN code here !

Bounding Box regression: -

It is step where.
A Multi Latered Perceptron is trying to minimize square error loss between predicted boxes and then original boxes.

Yes this is implemented in torchvision, you can read the docs here.

FasterRCNN does give you multiple loss outputs, regression loss, classification loss.
These are returned as metrics from train_step! (averaged to all images in epoch)

Have a look here

Thanks a lot for asking. Hope you like the library.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
question Further information is requested
Projects
None yet
Development

No branches or pull requests

2 participants