Skip to content

qdclone/bicycle_rl_control

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

bicycle_rl_control

This repository contains a reinforcement learning-based controller for bicycle balancing and navigation in simulation. The project uses the Genesis simulator and implements PPO (Proximal Policy Optimization) for training bicycle control policies.

genesis_bicycle.mp4

Installation

At first, install Genesis via PyPI:

pip install genesis-world

This project use the PPO implementation from rsl-rl to train the policy. Follow these installation steps:

# Install rsl_rl.
git clone https://github.com/leggedrobotics/rsl_rl
cd rsl_rl && git checkout v1.0.2 && pip install -e .

# Install tensorboard.
pip install tensorboard

Training

Train the bicycle policy using the BicycleEnv environment.

Run with:

python scripts/bicycle_train.py -e bicycle-policy -B 8192 --max_iterations 300

Train with visualization:

python scripts/bicycle_train.py -e bicycle-policy -B 8192 --max_iterations 300 -v

Evaluation

Evaluate the trained bicycle policy.

Run with:

python scripts/bicycle_eval.py -e bicycle-policy --ckpt 300 --record

Note: If you experience slow performance or encounter other issues during evaluation, try removing the --record option.

Reference

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%