This repository contains a modified version of the official Limelight Vision object detection training notebook, specifically adapted for Kaggle environments rather than Google Colab.
The Limelight Vision system is commonly used in FIRST Tech Challenge (FTC) competitions for computer vision and object detection tasks. This project provides tools to train custom detection models that can be deployed to Limelight hardware.
- Train custom object detection models for FTC game elements
- Supports TensorFlow 2.x with SSD MobileNet v2 architecture
- Generates optimized TFLite models compatible with Limelight hardware
- Includes quantized models (8-bit) and Google Coral TPU optimized versions
- Kaggle-specific adaptations for better performance and compatibility
- Kaggle notebook environment with GPU acceleration
- TensorFlow 2.15
- Python 3.10+
- Dataset in TFRecord format (can be created using tools like Roboflow)
- Create a new Kaggle notebook
- Upload the
limelight-detection.ipynbnotebook - Enable GPU acceleration in Kaggle notebook settings
- Run the cells in sequence
Your dataset should be prepared in TFRecord format with the following structure:
- Train folder with training examples and label map
- Valid folder with validation examples
- Test folder (optional)
The notebook guides you through:
- Installing the TensorFlow Object Detection API
- Setting up the training environment
- Loading and analyzing your dataset
- Configuring the model
- Training the model
- Converting to TFLite format
- Quantizing for better performance
- Compiling for Google Coral TPU (used in Limelight hardware)
After training completes, the notebook will generate:
- Full TensorFlow saved model
- 32-bit TFLite model (highest accuracy)
- 8-bit quantized TFLite model (better performance)
- Google Coral TPU optimized model (for Limelight hardware)
- Labels file
This adaptation differs from the official Limelight training notebook in these ways:
- Configured specifically for Kaggle rather than Google Colab
- Fixed compatibility issues with TensorFlow 2.15
- Modified paths and environment variables for Kaggle
- Improved error handling and progress reporting
- Added pre-training dataset validation
This project is adapted from the official Limelight Vision training notebook with modifications for Kaggle compatibility.
Please refer to the original Limelight Vision license for usage terms.