From a high level, in order to train our custom object detection model, we take the following steps in the Colab Notebook to Train TensorFlow Lite Model: Install TensorFlow object detection library and dependencies . Import dataset from Roboflow in TFRecord format . Write custom model configuration .