/toolpath-design-rl

Design toolpath for additive manufacturing using reinforcement learning

Primary LanguageJupyter NotebookMIT LicenseMIT

Toolpath Design for Additive Manufacturing using Reinforcement Learning

Using Muzero algorithm to design toolpaths for additive manufacturing process given any section geometry. Currently, we use a dense reward structure where a positive reward is assigned to correct material deposition, a negative reward is assigned to wrong material deposition, and a small negative reward is assigned to other motions at each time step. Significant parts of this code are adopted based on muzero-general.

To Dos

  • Accelerate the code
  • Dense reward structure test
  • Sparse reward structure test
  • Document hyper-parameter analysis
  • Pretraining networks

Getting started

Installation

git clone https://github.com/mojtabamozaffar/toolpath-design-rl
cd toolpath-design-rl

pip install -r requirements.txt

Usage

python main.py

License

This project is released under the MIT License.