data/raw download from https://github.com/jianzhang96/MSD ...Use google for a free limited GPU service
This application uses computer vision to detect defects on phone surfaces such as oil smudges, screen scratches, and stains.
- Clone this repository
- Install dependencies:
pip install -r requirements.txt
- Organize your dataset:
python src/data_preparation/organize_dataset.py
- Train the model:
python src/model/train_model.py
- Run the application:
python src/app/phone_defect_app.py
data/
: Contains raw and processed datasetsmodels/
: Stores trained model filessrc/
: Source code for data preparation, model training, and the main applicationdocs/
: Project documentationtests/
: Unit tests and integration tests
- Launch the application
- Click "Select Image" to choose a phone image
- Click "Detect Defect" to analyze the image
The application will display the detected defect type and confidence level.