ClassifyDefectMap_MixedWM38 - GitLite

KO | EN

gitlite — search

#large-language-models #gemini #terminal #editor #ui #security #electron #github #containers #claude-code

ClassifyDefectMap_MixedWM38

(★ 11)

Classification of wafer defect map patterns

File Explorer

ZIP Download

- - args.yaml
  - confusion_matrix.png
  - confusion_matrix_normalized.png
  - results.csv
  - results.png
  - train_batch0.jpg
  - train_batch1.jpg
  - train_batch2.jpg
  - val_batch0_labels.jpg
  - val_batch0_pred.jpg
  - val_batch1_labels.jpg
  - val_batch1_pred.jpg
  - val_batch2_labels.jpg
  - val_batch2_pred.jpg
- - args.yaml
  - class_indices.yaml
  - confusion.csv
  - confusion_matrix.png
  - confusion_matrix_normalized.png
  - custom_confusion_matrix.xlsx
  - metrics-OLD.yaml
  - metrics.yaml
  - model_structure-2.png
  - model_structure-3.png
  - model_structure.png
  - result_confusion_matrix.png
  - results.csv
  - results.png
  - tp_fp_fn_counts.yaml
  - train_batch0.jpg
  - train_batch1.jpg
  - train_batch2.jpg
  - val_batch0_labels.jpg
  - val_batch0_pred.jpg
  - val_batch1_labels.jpg
  - val_batch1_pred.jpg
  - val_batch2_labels.jpg
  - val_batch2_pred.jpg
- - args.yaml
  - confusion_matrix.png
  - confusion_matrix_normalized.png
  - results.csv
  - results.png
  - train_batch0.jpg
  - train_batch1.jpg
  - train_batch2.jpg
  - val_batch0_labels.jpg
  - val_batch0_pred.jpg
  - val_batch1_labels.jpg
  - val_batch1_pred.jpg
  - val_batch2_labels.jpg
  - val_batch2_pred.jpg
.gitignore
encodings.txt
encodings.yaml
MixedWM38_Kaggle.pdf
README.md
val_and_results.py
Wafer Map.png
wafer_defect_map.py
yolo_wafer_classification.py

# Use via CDN

jsDelivr

jsDelivr serves any public GitHub repository as a CDN with zero setup. Pick a version and a file to get a ready-to-paste link and snippet.

Version

File

Link

Example

// repository documentation

Was this content helpful?

★ 0 (0 ratings)

This is a demo project built for personal use using the [MixedWM38 dataset](https://github.com/Junliangwangdhu/WaferMap). Note that there is an issue with the dataset as pointed out in [this issue](https://github.com/Junliangwangdhu/WaferMap/issues/2), which was corrected for the results shared here.

## Wafer map patterns
![image](https://raw.githubusercontent.com/Burhan-Q/ClassifyDefectMap_MixedWM38/master/Wafer%20Map.png)

# Model

Uses the Ultralytics YOLOv8-Large classification model, with standard pretrained weights. The training was run for a short 10 epochs as this was only as a demo project.

# Results

## Confusion Matrix Result
![image](https://raw.githubusercontent.com/Burhan-Q/ClassifyDefectMap_MixedWM38/master/wafer_defects/EXP00002/result_confusion_matrix.png)

## Loss and Accuracy Plots
![image](https://raw.githubusercontent.com/Burhan-Q/ClassifyDefectMap_MixedWM38/master/wafer_defects/EXP00002/results.png)

Overall results from `EXP0002` which was a full GPU training with validation experiment, see the `args.yaml` file to view configuration. Additional metrics were computed using [val_and_results.py](/val_and_results.py). This result should not be considered complete, as model should be trained for additional epochs and additional hyperparameters explored. It is merely a demonstration of implementing a classifier model on wafer defect maps.