Object Detection Models on TensorFlow 2

Prerequsite

Train RetinaNet on TPU

Train a vanilla ResNet-50 based RetinaNet.

Train a SpineNet-49 based RetinaNet.

Train a custom RetinaNet using the config file.

Train RetinaNet on GPU

Train Mask R-CNN on TPU

Train a vanilla ResNet-50 based Mask R-CNN.

Train a SpineNet-49 based Mask R-CNN.

Train a custom Mask R-CNN using the config file.

Train Mask R-CNN on GPU

Train ShapeMask on TPU

Train a ResNet-50 based ShapeMask.

Train a custom ShapeMask using the config file.

Train ShapeMask on GPU

Run the evaluation (after training)

References