CS525: Reinforcement Learning - Worcester Polytechnic Institute, Fall 2022
Members: Rutwik Bonde, Prathamesh Bhamare, Aniket Patil - Master of Science in Robotics Engineering
Our goal is to train RL agents to navigate ego vehicle safely within racetrack-v0
environment, third party environment in the Open-AI gym and benchmark the results for lane keeping and obstacle avoidance tasks.
-
Clone this repository.
-
Create an empty python environment:
python3 -m venv rl_venv
- Activate the environment and install libraries
source rl_venv/bin/activate
pip3 install -r requirements.txt
- Configure the paramters in the config/params.yaml file.
IMPORTANT: For each run, ensure that you modify the
exp_id
so that the log and checkpoint files are not overwritten. Use theagent
tag to select between "PPO", "DDPG" and "A3C" agents.
- Run the
main.py
file
python3 main.py
-
In the same environment as above, change the config/params.yaml file by setting
train
tag to False. This will set the mode to Testing. Set thesave_video
tag to True to generate the video output for testing. -
Add path to the model in
load_model
. Use the models in the models folder to load pretrained weights.