This app let's you drag and drop an image of a car and it will tell you what kind of damage it has. The model is trained on third quarter front and rare view hence the picture should capture the third quarter front or rare view of a car.
- Used ResNet50 for transfer learning
- Model was trained on around 1700 images with 6 target classes
- Front Normal
- Front Crushed
- Front Breakage
- Rear Normal
- Rear Crushed
- Rear Breakage
- The accuracy on the validation set was around 80%
-
To get started, first install the dependencies using:
pip install -r requirements.txt -
Run the streamlit app:
streamlit run app.py
