Rocket Detect Computer Vision Project
Rocket Detect is the dataset used to train the nueral network behind Autotrack, NASASpaceflight's automated rocket tracking system. Rocket Detect uses three classes:
- Engine Flames - The fire produced by the rocket
- Rocket Body - The body of the launch vehicle
- 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 = { jun },
note = { visited on 2024-04-26 },
}
Connect Your Model With Program Logic
Find utilities and guides to help you start using the Rocket Detect project in your project.