rkshiyaniya / 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.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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.

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.

About

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.


Languages

Language:Jupyter Notebook 100.0%