Skip to content

helloankur/pcb-defect-detect-yolov5

Repository files navigation

PCB Defect Detect



Actual.Vs.Predict.mp4

This Repository present the Create Data set for train Model , how to train by using YoloV5 and test data by compare actual label and predict label. Create main.py to run the whole model and detect the defect from PCB Image.

Image Size required = 640 x 640

📝 yoloV5 :

https://github.com/ultralytics/yolov5

📁 ~~ Dataset :~~

https://github.com/tangsanli5201/DeepPCB

⤵️ Download Model and Weight

  • Download weight from weight Directory follow the link below
  • Download or clone YoloV5 model from above link.

🖥️ 🛠️ Download Required Library:

➡️ matplotlib>=3.2.2

➡️ numpy>=1.18.5

➡️ opencv-python>=4.1.2

➡️ Pillow>=8.0.0

➡️ PyYAML>=5.3.1

➡️ scipy>=1.4.1

➡️ torch>=1.7.0

➡️ torchvision>=0.8.1

➡️ tqdm>=4.41.0

🚀 Weights & Biases

bonous of W&B use for monitor live performance of your model.

by using command :

!pip install -q --upgrade wandb
### # Login 
import wandb
wandb.login()

use above code in Notebook ---> Jupyter,Kaggle or Google colab

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages