Skip to content

Latest commit

 

History

History
72 lines (39 loc) · 1.76 KB

README.md

File metadata and controls

72 lines (39 loc) · 1.76 KB

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