https://github.com/lspahija/simple-defect-detection

Super simple defect detection with PyTorch
https://github.com/lspahija/simple-defect-detection

Last synced: 3 days ago
JSON representation

Super simple defect detection with PyTorch

• Host: GitHub
• URL: https://github.com/lspahija/simple-defect-detection
• Owner: lspahija
• License: mit
• Created: 2024-01-20T00:38:49.000Z (11 months ago)
• Default Branch: main
• Last Pushed: 2024-01-21T17:38:14.000Z (11 months ago)
• Last Synced: 2024-11-01T12:09:07.232Z (about 2 months ago)
• Language: Jupyter Notebook
• Size: 400 MB
• Stars: 3
• Watchers: 2
• Forks: 0
• Open Issues: 0
• Metadata Files:
  • Readme: README.md
  • License: LICENSE

Awesome Lists containing this project

README

        
# Simple Defect Detector

## Overview

Product defect detection can be easy.

This repo shows how to fine-tune a ResNet-18 model to identify product defects. The included dataset is of cast submersible pump impellers, but other datasets can easily be dropped in, and the model finetuned on those instead.

Vanilla PyTorch is used and the training script is super simple.

## Find this useful?

Please star the repository!