Heyo!
I'm a CS grad and I had developed this same project in Octave for an online course.
So, I do have experience with this kind of problem.
I would like to know more about the kind of images that will be supplied
you can give me this information over chat and then we can take it forward from there.
Have a great day! :)