Skip to content

Object detection project for the RoboCup SPL team HULKs and Dutch Nao Team e.V.

Notifications You must be signed in to change notification settings

IntelligentRoboticsLab/robot_detection

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Object detection using a Single Shot Detector

In this project, my goal is to train a neural network to detect both robots and balls in the context of RoboCup SPL. I have implemented a Single Shot Detector (SSD) following the work of Liu et al.1. The architecture of the initial network is based on the works of BHuman. However, the implementation of BHuman is limited to detecting only Robots. This project is my attempt to extend the network such that it detects both robots and balls.

The training data for this project was obtained from RoboEireann2. A sample of nine images is provided below.

Setup

To set up the project, follow these steps:

  1. Clone the repository to your local machine.
git clone https://github.com/jonathan-hellwig/robot-detection.git
  1. Navigate to the project directory.
cd robot-detection
  1. Create a virtual environment and activate it.
python3 -m venv .venv
source .venv/bin/activate
  1. Install the required packages.
pip install -r requirements.txt

Training

The files robot_detection_train.ipynb and synthetic_data_train.ipynb contain the code to train a JetNet on the RoboEireann data and synthetic data respectively. Each training run automatically generates a directory with log files that can be visualized with tensorboard.

Footnotes

  1. SSD: Single Shot MultiBox Detector

  2. Faster YOLO-LITE: Faster Object Detection on Robot and Edge Devices

About

Object detection project for the RoboCup SPL team HULKs and Dutch Nao Team e.V.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Jupyter Notebook 99.7%
  • Python 0.3%