Problems to put my dataset in the same format used by this Retinanet

TensorFlow
, , ,
1

Hello,

I’m running this Keras implementation of Retinanet and it works perfectly on COCO dataset obtained from tfds.load().

How to run this model on my own object detection dataset?

Thanks in advance,
Allan Freitas

3

Hi @Allansdefreitas ,

This tutorial fine-tunes a RetinaNet with ResNet-50 as backbone model from the TensorFlow Model Garden for custom object detection dataset.
Object detection with Model Garden  |  TensorFlow Core.

And also you can follow the official TensorFlow Model Garden Github for more pre-trained models.

I Hope this helps!

Thanks