Rocket Detect

Object Detection


Rocket Detect Computer Vision Project

Drop an image or


12303 images
Explore Dataset

Rocket Detect is the dataset used to train the nueral network behind Autotrack, NASASpaceflight's automated rocket tracking system. Rocket Detect uses three classes:

  1. Engine Flames - The fire produced by the rocket
  2. Rocket Body - The body of the launch vehicle
  3. Space - A tiny spec in the sky that is the rocket after it has ascended into space

Using YoloV5 and Strongsort, Rocket Detect in its current form has proven sufficient to reliably track a Falcon 9 first stage continuously form launch to landing. It also works on a wide variety of other launch vehicles. The dataset will increase with time to further improve reliablity.

Trained Model API

This project has a trained model available that you can try in your browser and use to get predictions via our Hosted Inference API and other deployment methods.

Cite This Project

If you use this dataset in a research paper, please cite it using the following BibTeX:

                            title = { Rocket Detect Dataset },
                            type = { Open Source Dataset },
                            author = { NASASpaceflight },
                            howpublished = { \url{ } },
                            url = { },
                            journal = { Roboflow Universe },
                            publisher = { Roboflow },
                            year = { 2023 },
                            month = { jun },
                            note = { visited on 2024-02-21 },

Connect Your Model With Program Logic

Find utilities and guides to help you start using the Rocket Detect project in your project.

Last Updated

8 months ago

Project Type

Object Detection




Engine Flames Rocket Body Space

Views: 2253

Views in previous 30 days: 58

Downloads: 80

Downloads in previous 30 days: 0


CC BY 4.0