Skip to content

Developed an AI-powered defect detection system using Python, Flask, OpenCV, Scikit-learn, KMeans, HOG, and CLAHE under the NDE 4.0 framework. It enhances quality control in ultrasonic polymer welding through automated defect identification, contrast enhancement, and ML-based classification.

Notifications You must be signed in to change notification settings

NEHA-AMIN/AI-Defect-Detection-UPW

Folders and files

NameName
Last commit message
Last commit date

Latest commit

Β 

History

1 Commit
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 

Repository files navigation

Defect Detection Using Autoencoders and FCM

Overview

This project implements defect detection using Autoencoders and Fuzzy C-Means (FCM) clustering instead of traditional K-Means clustering. The model is designed to classify images into defected and non-defected categories.

Features

  • Uses Autoencoders for feature extraction.
  • Employs FCM clustering for classification.
  • Supports image-based defect detection.
  • Flask web interface for uploading and analyzing images.

Repository Structure

πŸ“‚ project_root/
 β”œβ”€β”€ πŸ“‚ templates/            # HTML templates for UI
 β”œβ”€β”€ πŸ“„ main.py               # Main application logic
 β”œβ”€β”€ πŸ“„ README.md             # Documentation
 β”œβ”€β”€ πŸ“¦ Defected.zip          # Dataset: Defected images
 β”œβ”€β”€ πŸ“¦ Non_defected.zip      # Dataset: Non-defected images

Requirements

Install dependencies using:

pip install -r requirements.txt

Usage

  1. Extract datasets from Defected.zip and Non_defected.zip.
  2. Run the Flask app:
    python main.py
  3. Access the Web Interface: Open http://127.0.0.1:5000/ in your browser.

Algorithm: Autoencoders + FCM

Instead of K-Means, this project uses Fuzzy C-Means (FCM) for clustering defected vs. non-defected images. This allows for a soft clustering approach, making the model more robust compared to hard-assignment methods like K-Means.

About

Developed an AI-powered defect detection system using Python, Flask, OpenCV, Scikit-learn, KMeans, HOG, and CLAHE under the NDE 4.0 framework. It enhances quality control in ultrasonic polymer welding through automated defect identification, contrast enhancement, and ML-based classification.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published