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.1.tar.gz (1.0 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.1-py3-none-any.whl (1.2 MB view details)

Uploaded Python 3

File details

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

File metadata

  • Download URL: tensoraerospace-0.3.1.tar.gz
  • Upload date:
  • Size: 1.0 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.1.tar.gz
Algorithm Hash digest
SHA256 af7c1f3dc34928b5e017206e637da5e850bb09a94bcd3c65d3679582c398f545
MD5 9e2354f46561d9b76c4f7096018d6bab
BLAKE2b-256 59c474d33dc4fe40182d3e2d95aa7ba8492e6687c602d8eb2c8b3bea23dee983

See more details on using hashes here.

File details

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

File metadata

  • Download URL: tensoraerospace-0.3.1-py3-none-any.whl
  • Upload date:
  • Size: 1.2 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.1-py3-none-any.whl
Algorithm Hash digest
SHA256 83cd3fe5485100154044905a516f478ac583a53c30b2cd47b43132285e6e9326
MD5 055734a2f65e4b45c657cd771c94d08e
BLAKE2b-256 19f3a591fb7c551046ee511c7df84a75be0298e7f4d05e7e67b93acf15825b52

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