Implementation of CNN in Keras for detecting anomalies in images of computer pins. Developed during my Internship at VSB in collaboration with Vitesco.
- Create new virtual environment
- Install packages from requirements.txt --> pip install -r requirements.txt
- Create data/ folder and insert subfolders with images of diffrent pin types
- Run preprocessing.py to generete data and labels
- Run main.py for trainning and testing