CNN based fabric defect classification for FabVis at UOM

Task

Implement a defect detection pipeline for Fabrics using Machine learning technique instead of YOLO.

Dataset

Available here : link

Repository Structure

dataset/
├── train/
│   ├── images/        # Training images
│   └── labels/        # YOLO-format labels
│
└── test/
|    ├── images/        # Test imagesRC
|    └── labels/        # YOLO-format labels
├── train/
│   ├── images/        # Training images
│   └── labels/        # YOLO-format labels
│___ RCNNDetect.py     # Training 
|___ RCNN.py           # RCNN model implementation 
|___ RCNNDataset.py    # Load Dataset 

Current Direction of the project

Now we are collecting a Dataset for every fabric defect. Stay touch for more details or contact me through here