Image Classification
Scikit-learn
English
defect-detection
machine-learning
quality-control
ensemble-learning
neural-networks
Eval Results (legacy)
Instructions to use AbdullahImran/Paper-Defect-Detection with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Scikit-learn
How to use AbdullahImran/Paper-Defect-Detection with Scikit-learn:
from huggingface_hub import hf_hub_download import joblib model = joblib.load( hf_hub_download("AbdullahImran/Paper-Defect-Detection", "sklearn_model.joblib") ) # only load pickle files from sources you trust # read more about it here https://skops.readthedocs.io/en/stable/persistence.html - Notebooks
- Google Colab
- Kaggle
| language: | |
| - en | |
| tags: | |
| - defect-detection | |
| - image-classification | |
| - machine-learning | |
| - quality-control | |
| - ensemble-learning | |
| - neural-networks | |
| datasets: | |
| - custom_paper_surface_defect | |
| pipeline_tag: image-classification | |
| model-index: | |
| - name: Paper Defect Detection | |
| results: | |
| - task: | |
| type: image-classification | |
| name: Surface Defect Detection | |
| metrics: | |
| - type: accuracy | |
| value: 0.81 | |
| name: Ensemble Test Accuracy | |
| - type: f1 | |
| value: 0.8 | |
| name: F1 Score | |
| library_name: sklearn | |
| metrics: | |
| - accuracy | |
| # Paper Defect Detection | |
| ## Model Description | |
| This model is designed for automated surface defect detection in manufacturing using a hybrid approach that combines classical machine learning and deep learning techniques. | |
| ### Model Architecture | |
| The model uses a hybrid architecture combining: | |
| - Logistic Regression | |
| - SVM | |
| - Naive Bayes | |
| - CNN | |
| - Ensemble Voting Classifier | |
| ### Feature Extraction Methods | |
| - Histogram of Oriented Gradients (HOG) | |
| - Gabor Filters | |
| - Canny Edge Detection | |
| - Wavelet Transforms | |
| ## Performance | |
| | Model | Train Accuracy | Test Accuracy | | |
| |--------------------|----------------|---------------| | |
| | Logistic Regression| 0.99 | 0.79 | | |
| | SVM | 0.86 | 0.80 | | |
| | Ensemble Model | 0.90 | 0.81 | | |
| ## Limitations | |
| - Performance may degrade for defect types not represented in the training data | |
| - Variations in lighting or textures can affect classification accuracy | |
| - This was a university project with room for improvement |