Repository contains ai training code & infer gradio for the plastic project. refer to the original repository
- Plastic_LeNet5.ipynb - a LeNet-5 model with Xavier initialization and 250 epoch technique from University of Amsterdam
- Plastic_infer.ipynb - a gradio interface to use the trained model and view the prediction of the class