SSDT：A single-shot detector for PCB defects

pytorch, SSDT, PCB defects detection

Prerequisites

Python 3.6
Numpy
Pytorch 0.3
torchvision
visdom
tqdm

Train dataset

support VOC, COCO and PCB dataset (proposed by PKU-HRI)

Train the model

python train_test.py

Use the visdom

python -m visdom.server -port ****

Result examples