Accompanying code to the publication: "Quantitative and Real-Time Control of 3D Printing Material Flow Through Deep Learning"
This repository allows you to easily train a regnet neural network to quantitatively predict with regression the relative state of the material flow rate in real time during printing from a single input image.
First create a Python 3 virtual environment and install the requirements - this should only take a couple of minutes. In the paper we used PyTorch (v1.10.1), Torchvision (v0.11.1), and CUDA (v11.3) in this work. You can use later versions if required and this repo has since been updated. See the complete list of requirements in requirements.txt.
virtualenv -p python3 env
source env/bin/activate
pip install -r requirements.txt
Inside the src directory are two sub-directories for our data and model. We use Pytorch-Lightning (v1.1.4) as a wrapper for both the dataset and datamodule classes and for our model.
python train.py