This repository contains a notebook for object detection with the help of fine tuning of a (TF2 friendly) RetinaNet architecture on very few examples of a novel class after initializing from a pre-trained COCO checkpoint. Training runs in eager mode.
Tech. Stack :
- Python
- TensorFlow/Keras
- object_detection API
- Matplotlib
- NumPy
- Other CV related libraries
Categories :
- Object Detection (Computer Vision)
- Deep Learning
- Image Annotation
- Eager Mode training
- Fine Tuning
- Detailed explaination has been presented in the notebook itself.