Metal Crack Detector
An Image classifier build with CNN to detect cracks and wrinkle on metal surface.
Prerequisites
Hardware and software requirements are listed bellow
- PC with (at least) 2GB GPU.
- Install CUDA 10, as its required for TensorFlow 2.0.
Installation
- Create a Virtual environment
- Install dependencies (
pip install -r requirements.txt
) - Run the code (
python3 train.py
)