Create RetinaNet object detection guide #2069
Create RetinaNet object detection guide #2069divyashreepathihalli merged 15 commits intokeras-team:masterfrom
Conversation
divyashreepathihalli
left a comment
There was a problem hiding this comment.
Thanks for the PR @sachinprasadhs - since this is a guide - we should have more detailed explanations that would help users understand our framework better. Lets add more of that.
Thanks!
mattdangerw
left a comment
There was a problem hiding this comment.
Thanks! All minor comments. Also heads up I added keras-team/keras-hub#2219 for coco id - name mappings. Whenever it is released we can use it.
|
/gemini review |
|
@gemini-code-assist review |
There was a problem hiding this comment.
Code Review
This pull request adds a new tutorial for object detection using RetinaNet and KerasHub, covering data preparation, inference, and fine-tuning. The implementation includes notebook, script, and markdown versions. Reviewers identified several issues, most notably a bug in the metadata builder that caused data leakage by loading all annotations instead of split-specific ones. Other feedback included a pathing error in the VOC data loader, the use of global variables in preprocessing functions, and a recommendation to separate backup and checkpoint directories for better file management. All suggested improvements aim to enhance the correctness and modularity of the tutorial code.
Summary:
This Pull Request introduces comprehensive documentation detailing how to use an object detection model from Keras Hub, using RetinaNet as an example.
This tutorial includes: