One of the most interesting projects I've worked on in the past couple of years was a project about image processingimage processing. The goal was to develop a system to be able to recognize Coca-Cola cans'cans' (note that I'm stressing the word cans'cans', you'll see why in a minute). You can see a sample below, with the can recognized in the green rectanglegreen rectangle with scale and rotation.
- The background could be very noisy.
- The cancan could have any scalescale or rotationrotation or even orientation (within reasonable limits).
- The image could have some degree of fuzziness (contours might not be entirely straight).
- There could be Coca-Cola bottles in the image, and the algorithm should only detect the cancan!
- The brightness of the image could vary a lot (so you can't rely "too much" on color detection).
- The cancan could be partly hidden on the sides or the middle and possibly partly hidden behind a bottle.
- There could be no canscan at all in the image, in which case you had to find nothing and write a message saying so.
Language: Done in C++ using OpenCVOpenCV library.
- Changing color domain from RGB to HSV (Hue Saturation ValueHSV) and filtering based on "red" hue, saturation above a certain threshold to avoid orange-like colors, and filtering of low value to avoid dark tones. The end result was a binary black and white image, where all white pixels would represent the pixels that match this threshold. Obviously there is still a lot of crap in the image, but this reduces the number of dimensions you have to work with).
- Noise filtering using median filtering (taking the median pixel value of all neighbors and replace the pixel by this value) to reduce noise.
- Using Canny Edge Detection Filter to get the contours of all items after 2 precedent steps.
Algorithm: The algorithm itself I chose for this task was taken from this (awesome) book on feature extractionthis awesome book on feature extraction and called Generalized Hough Transform (pretty different from the regular Hough Transform). It basically says a few things:
In the end, you end up with a heat map of the votes, for example here all the pixels of the contour of the can will vote for its gravitational center, so you'll have a lot of votes in the same pixel corresponding to the center, and will see a peak in the heat map as below.:
- It is extremely slow! I'm not stressing this enough. Almost a full day was needed to process the 30 test images, obviously because I had a very high scaling factor for rotation and translation, since some of the cans were very small.
- It was completely lost when bottles were in the image, and for some reason almost always found the bottle instead of the can (perhaps because bottles were bigger, thus had more pixels, thus more votes)
- Fuzzy images were also no good, since the votes ended up in pixel at random locations around the center, thus ending with a very noisy heat map.
- InvarianceIn-variance in translation and rotation was achieved, but not in orientation, meaning that a can that was not directly facing the camera objective wasn't recognized.
Can you help me improve my specific algorithm, using exclusively OpenCV OpenCV features, to resolve the four specific issues mentioned?
I hope some people will also learn something out of it as well, after all I think not only people who ask questions should learn. :)