Skip to main content

Open source deep learning framework that focuses on aerospace objects (rockets, planes, UAVs)

Project description

๐Ÿš€ TensorAeroSpace

en ru Documentation Status Python License GitHub stars Coverage Status TensorAeroSpace Logo

Advanced Aerospace Control Systems & Reinforcement Learning Framework

A comprehensive Python library for aerospace simulation, control algorithms, and reinforcement learning implementations

๐Ÿ“– Documentation โ€ข ๐Ÿš€ Quick Start โ€ข ๐Ÿ’ก Examples โ€ข ๐Ÿค Contributing


๐ŸŒŸ Overview

TensorAeroSpace is a cutting-edge Python framework that combines aerospace engineering with modern machine learning. It provides:

  • ๐ŸŽฏ Control Systems: Advanced control algorithms including PID, MPC, and modern RL approaches
  • โœˆ๏ธ Aerospace Models: High-fidelity aircraft and spacecraft simulation models
  • ๐ŸŽฎ OpenAI Gym Integration: Ready-to-use environments for reinforcement learning
  • ๐Ÿง  RL Algorithms: State-of-the-art reinforcement learning implementations
  • ๐Ÿ”ง Extensible Architecture: Easy to extend and customize for your specific needs

๐Ÿš€ Quick Start

๐Ÿ“ฆ Installation

Using Poetry (Recommended)

git clone https://github.com/tensoraerospace/tensoraerospace.git
cd tensoraerospace
poetry install

Using pip

pip install tensoraerospace

๐Ÿณ Docker

docker build -t tensoraerospace . --platform=linux/amd64
docker run -v $(pwd)/example:/app/example -p 8888:8888 -it tensoraerospace

๐Ÿƒโ€โ™‚๏ธ Quick Example

import tensoraerospace as tas

# Create an F-16 environment
env = tas.envs.F16Env()

# Initialize a PPO agent
agent = tas.agent.PPO(env.observation_space, env.action_space)

# Train the agent
for episode in range(1000):
    obs = env.reset()
    done = False
    while not done:
        action = agent.act(obs)
        obs, reward, done, info = env.step(action)

๐Ÿค– Supported Algorithms

Algorithm Type HuggingFace Export Status
IHDP Incremental Heuristic Dynamic Programming โŒ โœ…
DQN Deep Q-Learning โŒ โœ…
DDPG Deep Deterministic Policy Gradient โŒ โœ…
SAC Soft Actor-Critic โœ… โœ…
A3C Asynchronous Advantage Actor-Critic โŒ โœ…
PPO Proximal Policy Optimization โœ… โœ…
GAIL Imitation Learning (Adversarial) โŒ โœ…
MPC Model Predictive Control โœ… โœ…
A2C Advantage Actor-Critic โœ… โœ…
A2C-NARX A2C with NARX Critic โŒ โœ…
PID Proportional-Integral-Derivative โœ… โœ…

โœˆ๏ธ Aircraft & Spacecraft Models

๐Ÿ›ฉ๏ธ Fixed-Wing Aircraft
  • General Dynamics F-16 Fighting Falcon - High-fidelity fighter jet model
  • Boeing 747 - Commercial airliner dynamics
  • McDonnell Douglas F-4C Phantom II - Military aircraft model
  • North American X-15 - Hypersonic research aircraft
๐Ÿš UAVs & Drones
  • LAPAN Surveillance Aircraft (LSU)-05 - Indonesian surveillance UAV
  • Ultrastick-25e - RC aircraft model
  • Generic UAV State Space - Configurable UAV dynamics
๐Ÿš€ Rockets & Satellites
  • ELV (Expendable Launch Vehicle) - Launch vehicle dynamics
  • Generic Rocket Model - Customizable rocket simulation
  • Geostationary Satellite - Orbital mechanics simulation
  • Communication Satellite - ComSat dynamics and control

๐ŸŽฎ Simulation Environments

๐ŸŽฏ Unity ML-Agents Integration

Unity Demo

TensorAeroSpace seamlessly integrates with Unity ML-Agents for immersive 3D simulations:

  • ๐ŸŽฎ 3D Visualization: Real-time 3D aircraft simulation
  • ๐Ÿ”„ Real-time Training: Train agents in realistic environments
  • ๐Ÿ“Š Rich Sensors: Camera, LiDAR, and physics-based sensors
  • ๐ŸŒ Custom Environments: Build your own aerospace scenarios

๐Ÿ“ Example Environment: UnityAirplaneEnvironment

๐Ÿ”ง MATLAB Simulink Support

Simulink Model

  • ๐Ÿ“ Model Import: Convert Simulink models to Python
  • โšก High Performance: Compiled C++ integration
  • ๐Ÿ”„ Bidirectional: MATLAB โ†” Python workflow
  • ๐Ÿ“Š Validation: Cross-platform model validation

๐Ÿ“Š State Space Matrices

Mathematical foundation for control system design:

  • ๐Ÿงฎ Linear Models: State-space representation
  • ๐ŸŽ›๏ธ Control Design: Modern control theory implementation
  • ๐Ÿ“ˆ Analysis Tools: Stability, controllability, observability
  • ๐Ÿ”„ Linearization: Nonlinear model linearization

๐Ÿ“š Examples & Tutorials

Explore our comprehensive example collection in the ./example directory:

Category Description Notebooks
๐Ÿš€ Quick Start Basic usage and concepts quickstart.ipynb
๐Ÿค– Reinforcement Learning RL algorithm implementations reinforcement_learning/
๐ŸŽ›๏ธ Control Systems PID, MPC controllers pid_controllers/, mpc_controllers/
โœˆ๏ธ Aircraft Models Environment examples environments/
๐Ÿ”ง Optimization Hyperparameter tuning optimization/

๐Ÿ› ๏ธ Development & Contributing

We welcome contributions! Please see our Contributing Guide for details.

๐Ÿ—๏ธ Development Setup

git clone https://github.com/tensoraerospace/tensoraerospace.git
cd tensoraerospace
poetry install --with dev
poetry run pytest  # Run tests

๐Ÿงช Testing

# Run all tests
poetry run pytest

# Run specific test category
poetry run pytest tests/envs/
poetry run pytest tests/agents/

๐Ÿ“– Documentation

  • ๐Ÿ“š Full Documentation: tensoraerospace.readthedocs.io
  • ๐Ÿš€ API Reference: Detailed API documentation
  • ๐Ÿ“ Tutorials: Step-by-step guides
  • ๐Ÿ’ก Examples: Practical use cases

๐Ÿค Community & Support

๐Ÿ“„ License

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

๐Ÿ™ Acknowledgments

  • OpenAI Gym team for the excellent RL framework
  • Unity ML-Agents team for 3D simulation capabilities
  • The aerospace engineering community for domain expertise
  • All contributors who make this project possible

โญ Star us on GitHub if you find TensorAeroSpace useful! โญ

Made with โค๏ธ by the TensorAeroSpace team

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

tensoraerospace-0.3.3.tar.gz (19.8 MB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

tensoraerospace-0.3.3-py3-none-any.whl (20.4 MB view details)

Uploaded Python 3

File details

Details for the file tensoraerospace-0.3.3.tar.gz.

File metadata

  • Download URL: tensoraerospace-0.3.3.tar.gz
  • Upload date:
  • Size: 19.8 MB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: poetry/2.2.1 CPython/3.10.18 Linux/6.11.0-1018-azure

File hashes

Hashes for tensoraerospace-0.3.3.tar.gz
Algorithm Hash digest
SHA256 2b948eb81f3613b21fb414af4641480c2a6fbfc5517a27c84238d700cbf9ddcc
MD5 dfc4590fe6481ceeaf56bd07573ab4d4
BLAKE2b-256 987328bbad9eaafb8efbfc594d7cf565b364ae1d4cb60d25a4f065780113c49d

See more details on using hashes here.

File details

Details for the file tensoraerospace-0.3.3-py3-none-any.whl.

File metadata

  • Download URL: tensoraerospace-0.3.3-py3-none-any.whl
  • Upload date:
  • Size: 20.4 MB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: poetry/2.2.1 CPython/3.10.18 Linux/6.11.0-1018-azure

File hashes

Hashes for tensoraerospace-0.3.3-py3-none-any.whl
Algorithm Hash digest
SHA256 275bdac1c0b5deb4fb88192f6e690ebd48ea0fe043426d3c7cf860299bb4506d
MD5 aa71ca11153f83e564602c1a52f26153
BLAKE2b-256 0e8a96ab5ac44ef1c3a70268fb9b8cea4df823aecc3dfc43660841df3daf3a52

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page