Rocket Detect Computer Vision Project

NASASpaceflight

Updated a year ago

5.2k

views

180

downloads
Classes (3)
Engine Flames
Rocket Body
Space

Metrics

Try This Model
Drop an image or
Description

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.

Use This Trained Model

Try it in your browser, or deploy via our Hosted Inference API and other deployment methods.

Supervision

Build Computer Vision Applications Faster with Supervision

Visualize and process your model results with our reusable computer vision tools.

Cite This Project

LICENSE
CC BY 4.0

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-11-21 },
                            }
                        
                    

Similar Projects

See More
400 images 1 model