VigneshBondugula / PCB-Defect-Segmentation

Experiment to achieve Few Shot Segmentation of the parts with defects on a PCB

PCB Defect Segmentation

Documentation:

https://www.notion.so/PCB-Defect-Segmentation-3dc0c3dd1f6f4422979d6b90ed79713b?pvs=4

Flow Chart :

Example:

How to run:

• First, download the dataset from, Defects Instance Segmentation Dataset

• Copy the downloaded folders into data folder and run the notebooks given in it to extract the masks. If only masks for short circuit defects need to be generated run dataset_shot_circ.ipynb. Generated train_final, val_final and test_final folders need to be renamed to train, val and test.

• Run the PCB-segmentation.ipynb, make sure you have all the necessary python libraries.