项目作者: zalkikar

项目描述 :
Behavior cloning for OpenAI Gym's CarRacing-v0 environment.
高级语言: Python
项目地址: git://github.com/zalkikar/behaviorCloning_CarRacingv0.git
创建时间: 2020-10-05T20:27:06Z
项目社区:https://github.com/zalkikar/behaviorCloning_CarRacingv0

开源协议:MIT License

下载


Behavior Cloning for OpenAI’s CarRacing-v0

  1. |-- drive_manually.py (with flag --collect_data, stores manual driving data in './data/')
  2. |-- model.py (agent in './models/')
  3. |-- train_agent.py (train agent)
  4. |-- test_agent.py (agent performance, stores results in './results/')

Install pyTorch and gym requirements (I used an anaconda Python3.6 virtual env).

Neural Network has a validation MSE loss of 0.06 and mean episode rewards of 450 using a current image input grayscaled to (1,96,96) and preprocessed.

  1. python test_agent.py

gives the following results:

Alt Text