search

s39674 / Image2schematic

Trying to extract pcb schematics from images using computer vision
37 stars 15 forks source link

PCB Component Detection #16

Closed s39674 closed 2 years ago

s39674 commented 2 years ago

The final schematic should include resistors, capacitors, inductors etc.. So we need a way to detect those component. There is a nice dataset of PCB components here: https://sites.google.com/view/chiawen-kuo/home/pcb-component-detection but I don't know how to start training a model, can anyone help? Thanks!

s39674 commented 2 years ago

UPDATE: I created https://github.com/s39674/PCB-Component-Detection with very beginner model similar to this example: https://pytorch.org/tutorials/beginner/blitz/cifar10_tutorial.html#sphx-glr-beginner-blitz-cifar10-tutorial-py ; I'm looking for feedback and suggestions.