This project implements a fabric defect detection system using Streamlit for a user-friendly interface and YOLOv8 for object detection.
Manual fabric defect inspection is time-consuming and prone to human error. This project addresses this challenge by offering an automated solution for fabric quality control.
- Upload fabric images for defect detection.
- Visualize the detected defects with labels.
- Python 3.6 or later
- Streamlit
- YOLOv8 (https://github.com/topics/yolov8)
- Clone this repository.
- Create a new virtual environment (recommended).
- Install required dependencies:
pip install streamlit yolov8
- Download pre-trained YOLOv8 weights (compatible with your dataset) and place them in the project directory.
- Run the application:
streamlit run main.py
- Access the application in your web browser at http://localhost:8501.
- Upload a fabric image and the model will predict and visualize any detected defects.