Skip to content

Tuanshu/SSDT-A-single-shot-detector-for-PCB--defects

 
 

Repository files navigation

SSDT:A single-shot detector for PCB defects

pytorch, SSDT, PCB defects detection

Prerequisites

  1. Python 3.6
  2. Numpy
  3. Pytorch 0.3
  4. torchvision
  5. visdom
  6. 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

About

pytorch, SSDT, PCB defects detection, for study purpose

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Languages

  • Python 99.3%
  • Other 0.7%