-
Notifications
You must be signed in to change notification settings - Fork 248
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
[RetinaNet] Image Converter and ObjectDetector #1906
[RetinaNet] Image Converter and ObjectDetector #1906
Conversation
…neeli/keras-hub into sineeli/add-retinanet-phase-2
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thanks for the PR @sineeli!! Looks generally good!! I have left a few comments.
Also, I want to make sure the -The code usage is updated to reflect the correct implementation - https://docs.google.com/document/d/15FUEP_vNehwLWJLragXhPFkYcbmmpbo0NYh5vL6q1xA/edit?tab=t.0
Kept some layers |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thanks @sineeli. I left a few comments.
"max_level": 7, | ||
"num_scales": 3, | ||
"aspect_ratios": [0.5, 1.0, 2.0], | ||
"anchor_size": 8, | ||
}, | ||
input_data={ |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
define input_data also in setup()
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
input_data varies in each test_case so kept it separately also how we assert changes
- Correct test cases.
…et format from preprocessor
… sineeli/add-retinanet-phase-2
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thanks Siva! added a few NIT comments! looks great!
This PR covers preprocessor for RetinaNet object detector and RetinaNet model itself. #1756
ImageObjectDetector
ImageObjectDetectorPreprocessor
RetinaNetObjectDetector
RetinaNetObjectDetectorPreprocessor
RetinaNetImageConverter