1

I want to create an AI to detect and identify certain edge lines on my image. The input image is a locker key, and I want to know the exact position of certain edges.

Sample input image:

enter image description here

Sample output image. As output i have each red line position.

enter image description here

I was thinking in finetuning some kind of instace segmentation or even yolo network but maybe there is a better approach. Ideally, I would like the software to be light enoguh to run in a mobilephone.

I also considered two "simpler" approaches:

  • Using corner detection
  • Using Unet to get the contour + opencv approxpoly to get straight lines coordinates

Nevertheless, I believe I would obtain a more consistent result with more advanced AI. (I do have a lot of training images to use)

Lluis C
  • 111
  • 3

1 Answers1

2

I don't think that more advanced AI would necessarily produce more consistent results. Check something as simple as the Prewitt operator, which is pretty damn good at edge detection. I would suggest starting with something simple, looking at what are the failure edge cases (pun unintended), and only use a more complicated network if the failure cases are plenty and addressable by some more advanced technique.

ImotVoksim
  • 84
  • 2