Augmentation of WM811K Silicon Wafer Map Dataset for Error Pattern Detection Training

Authors: David Guo 1007677275

Installation

All code is written in Python 3.8. Development was done on WSL2 in Ubuntu 20.04.

For install instructions, please see 'install.md

Data Pre-processing and Setup

To start, download the WM-811k Dataset. Please see 'main.ipynb' for further guidance.

Training and Generation

Please see 'main.ipynb' for all project code.

All training conducted on an RTX 3070 Ti 8gb, and a Ryzen 7 5800X3D with 32gb system RAM