/Eager-Few-Shot-Object-Detection

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.

Primary LanguageJupyter Notebook

Watchers