Search code examples
pythonopencvcomputer-visionsymbols

Symbol detection in scheme using Computer Vision


I am trying to detect electrical symbol in electrical scheme. Here I think 2 ways could be use:

  • classical way with OpenCV, I tried re to recognise shape with opencv and python but some symbole are too complexe
  • deep learning way: I tried with Mask-RCNN using a handmade dataset of symbol but nothing get really successful

Here is a really simple example of what I would like to do: enter image description here

I think it could be easy to make a dataset of symbol but all symbol would be the same form and context of the image would not be represented.

How do you think I could handle this problem ?


Solution

  • QATM:Quality-Aware Template Matching For Deep Learning might be what you are looking for.

    Original paper : https://arxiv.org/abs/1903.07254

    And the following github contain an example with electric scheme: https://github.com/kamata1729/QATM_pytorch

    enter image description here