-
Notifications
You must be signed in to change notification settings - Fork 45.6k
add CIOU/DIOU/GIOU losses (from efficientdet library) #9083
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
base: master
Are you sure you want to change the base?
Conversation
extending iou loss options
IOU functions
adding IOU losses
Note -- the autoaugment changes were upgrades for TF2 compatibility. They have also been verified (also taken from automl efficientdet repository), but not sure why they got added to this PR. I tried to separate them into a different pull request but no luck.... |
Also a note for a TODO for down the road....when using the autoaugment policy with augmix=True (default is False), the bbox returned on the transformations is not correct when image gets blended with rotates and translates. I didn't modify the code from the automl repository so its a carryover from there. |
I don't know if these could go in the new folder https://github.com/tensorflow/models/tree/master/official/vision/keras_cv/losses /cc @tanzhenyu |
Will this PR be checked? I think ciou is a great feature to add |
Description
Type of change
For a new feature or function, please create an issue first to discuss it
with us before submitting a pull request.
Note: Please delete options that are not relevant.
Tests
Test Configuration:
Checklist