Skip to content
Public template

About

Dive into VISTA! πŸ”­ This cutting-edge Object Detection Model is meticulously trained on the unique DualityAI Twins Falcon Space Station Dataset, ready to identify objects with stellar precision. πŸ›°οΈπŸŽ―

Topics

Resources

Stars

3 stars

Watchers

2 watching

Forks

Latest commit

Β 

History

61 Commits

Folders and files

NameName
Last commit message
Last commit date
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 

Repository files navigation

Python Version PyTorch Version License

VISTA-S: Visual Inference System for Target Assessment

DualityAI Space Station Model

Welcome to VISTA-S β€” a cutting-edge, high-precision AI system for object detection and visual inference.
Inspired by the mysteries of space, VISTA-S delivers powerful computer vision capabilities with ease and elegance. 🌌


✨ Features

  • State-of-the-Art Detection: Built on YOLOv8 for high-precision object detection.
  • Optimized Performance: Fast, efficient inference for real-time applications.
  • Interactive Demo: Visualize and interact with predictions through an easy-to-use web app.
  • Seamless Data Integration: Streamlined data preparation and handling for a smooth workflow.

πŸ“Έ VISTA-S in Action

Watch the video


⚑ Quickstart

Get up and running in minutes!

1. Create Environment

conda env create -f environment.yaml

2. Activate Environment

conda activate VISTA

πŸ“¦ Data Preparation

  • Download the Falcon Dataset:
    Download here

  • Unzip & Place:
    Extract the dataset and copy its contents to:

    data/raw/
    

Note: The dataset is NOT included in this repository due to its size. Please download it manually.


πŸ‹οΈβ€β™‚οΈ Training

Train the model on your machine:

python src/train.py

πŸ” Inference

Detect objects in a sample image:

python src/detect.py data/raw/test/images/sample.jpg

πŸ–₯️ Demo Application

Experience VISTA-S through the interactive web app:

  1. Navigate to the app directory:

    cd app
  2. Install requirements:

    pip install -r requirements.txt
  3. Start the backend server:

    python backend.py

πŸ“Š Performance

VISTA-S achieves exceptional results on the Falcon dataset:

  • Precision: ~0.9797
  • Recall: ~0.9088
  • mAP@0.5: ~0.9416
  • mAP@0.5:0.95: ~0.8843

These scores are based on the YOLOv8 architecture.
For detailed logs and more metrics, see the models/logs/yolov8_observo/ directory.


πŸ“ Project Structure

β”œβ”€β”€ app/                   # Flask backend app
β”‚   β”œβ”€β”€ backend.py
β”‚   β”œβ”€β”€ routes.py
β”‚   β”œβ”€β”€ simple_backend.py
β”‚   β”œβ”€β”€ requirements.txt
β”‚   └── templates/
β”‚       └── index.html
β”œβ”€β”€ config/
β”‚   └── observo.yaml
β”œβ”€β”€ data/
β”‚   └── raw/
β”‚       β”œβ”€β”€ classes.txt
β”‚       β”œβ”€β”€ predict.py
β”‚       β”œβ”€β”€ train.py
β”‚       β”œβ”€β”€ visualize.py
β”‚       β”œβ”€β”€ yolo_params.yaml
β”‚       └── data/
β”œβ”€β”€ docs/
β”‚   └── report_outline.md
β”œβ”€β”€ mobile/
β”‚   β”œβ”€β”€ src/
β”‚   β”‚   β”œβ”€β”€ App.js
β”‚   β”‚   β”œβ”€β”€ api/
β”‚   β”‚   └── screens/
β”‚   β”œβ”€β”€ package.json
β”‚   β”œβ”€β”€ README.md
β”‚   └── SETUP.md
β”œβ”€β”€ models/
β”‚   β”œβ”€β”€ weights/
β”‚   └── logs/
β”œβ”€β”€ notebooks/
β”‚   β”œβ”€β”€ EDA.ipynb
β”‚   └── train_yolov8.ipynb
β”œβ”€β”€ src/
β”‚   β”œβ”€β”€ detect.py
β”‚   β”œβ”€β”€ train.py
β”‚   β”œβ”€β”€ utils.py
β”‚   └── constraints.txt
β”œβ”€β”€ uploads/
β”œβ”€β”€ Web_App_frontend/
β”‚   β”œβ”€β”€ src/
β”‚   β”‚   β”œβ”€β”€ components/
β”‚   β”‚   β”œβ”€β”€ pages/
β”‚   β”‚   └── hooks/
β”‚   β”œβ”€β”€ package.json
β”‚   └── vite.config.ts
β”œβ”€β”€ environment.yaml
β”œβ”€β”€ requirements.txt
β”œβ”€β”€ requirements_minimal.txt
β”œβ”€β”€ render.yaml
β”œβ”€β”€ gunicorn_config.py
β”œβ”€β”€ Procfile
β”œβ”€β”€ .gitignore
β”œβ”€β”€ .gitattributes
β”œβ”€β”€ DEPLOYMENT.md
└── README.md

Vista Sample 1 Vista Sample 2

Vista Sample 3 Vista Sample 4

---

⚠️ DO NOT COMMIT SENSITIVE OR LARGE FILES

  • Model weights, logs, uploads, and raw data are excluded via .gitignore.
  • Do NOT commit files in models/weights/, models/logs/, uploads/, or data/raw/.
  • Notebooks and environment folders are also excluded.

🀝 Contributing

We welcome your contributions!

  1. Fork the repository.
  2. Create a branch for your changes.
  3. Submit a pull request with a clear description.

For major changes or new features, please open an issue first to discuss.


πŸ“„ License

This project is licensed under the MIT License.
See the LICENSE file for details.


✨ Explore the universe with VISTA-S!
Star the repo, open issues, or contribute to its growth.
Your feedback and contributions are always welcome.

About

Dive into VISTA! πŸ”­ This cutting-edge Object Detection Model is meticulously trained on the unique DualityAI Twins Falcon Space Station Dataset, ready to identify objects with stellar precision. πŸ›°οΈπŸŽ―

Topics

Resources

Stars

3 stars

Watchers

2 watching

Forks

Packages

Contributors

Languages