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
https://github.com/ultralytics/yolov5
https://github.com/tangsanli5201/DeepPCB
- Download weight from weight Directory follow the link below
- Download or clone YoloV5 model from above link.
➡️ 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
by using command :
!pip install -q --upgrade wandb
### # Login
import wandb
wandb.login()
use above code in Notebook ---> Jupyter,Kaggle or Google colab