A very simple PyTorch notebook project for training and evaluating a traffic sign classification model.
- loads a small subset of the GTSRB German Traffic Sign dataset
- applies image transforms and prepares training/validation data
- trains a convolutional neural network with PyTorch
- evaluates accuracy on a validation subset
blast-ai-capstone-project.ipynb- main notebook containing the data loading, training, and evaluation workflow
- The notebook uses
torch,torchvision,numpy, andpandas - It is intended as a simple capstone example rather than a production application