Rocket Detect Computer Vision Project

Drop an image or

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:

@misc{ rocket-detect_dataset,
    title = { Rocket Detect Dataset },
    type = { Open Source Dataset },
    author = { NASASpaceflight },
    howpublished = { \url{ https://universe.roboflow.com/nasaspaceflight/rocket-detect } },
    url = { https://universe.roboflow.com/nasaspaceflight/rocket-detect },
    journal = { Roboflow Universe },
    publisher = { Roboflow },
    year = { 2023 },
    month = { jan },
    note = { visited on 2023-01-26 },

Last Updated

8 days ago

Project Type

Object Detection




Engine Flames, Rocket Body, Space


CC BY 4.0