diff --git a/README.md b/README.md index 67c9766..85ff015 100644 --- a/README.md +++ b/README.md @@ -1,4 +1,5 @@ # Reinforcement Learning with Cozmo and Unity +@Author: Tobias Hassel This project uses Reinforcement Learning to teach a virtual Cozmo-Robot to follow a Line drawn on the ground.The plugin ML-Agents for Unity is used to train the robot. A camera together with OpenCvSharp identifies the Line. Based on the Center of Gravity the reward is calculated for the robot. After the training the result is a model file which can be used by the real robot. A python project controls the real cozmo using the pretrained model.