Automated Optical Inspection implemented by Python. It's assignment 2 in Introduction to Deep Learning Course.
cd training
python training.py
python test.py
conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=10.2 -c pytorch
https://pytorch.org/tutorials/beginner/transfer_learning_tutorial.html