A collection of Crazyflie reinforcement learning environments built on Isaac Lab. Trained policies can be deployed on real Crazyflie 2.1 drones via the scripts present in execution/.
Install Isaac Lab 2.3.2 (Isaac Sim 4.5+).
Install with the extras you need:
pip install -e "source/CrazyPlayGround[rl]" # Simulation / Training
pip install -e "source/CrazyPlayGround[deploy]" # Deployment Hardware
pip install -e "source/CrazyPlayGround[all]" python scripts/list_envs.py
python scripts/skrl/train.py --task=Vel-Hovering --num_envs=4096A Dockerfile is provided for reproducible training (Linux + NVIDIA GPU required):
docker login nvcr.io
docker build -t crazyplayground -f docker/Dockerfile .
docker run --gpus all -it --rm -v $(pwd)/logs:/workspace/crazyplayground/logs crazyplaygroundOnce inside the container, use the following command to verify and train (inside the container the --headless is mandatory):
/workspace/isaaclab/isaaclab.sh -p scripts/list_envs.py
/workspace/isaaclab/isaaclab.sh -p scripts/skrl/train.py --task=Vel-Hovering --num_envs=4096 --headlessFull documentation (environments, controller architecture, configuration, real-drone deployment) lives in docs/ and is built with MkDocs Material, an online version of the documentation will soon be available
