treadscan 0.5.0

Tools for scanning tire tread(s).

treadscan

This package provides utilities for extracting tire tread (pattern) from images, or even from video of passing cars.

Three main classes are
treadscan.Detector - picks out stopped cars from continoous footage (like at an intersection, under a traffic light). treadscan.Segmentor - finds ellipse defining car wheel in image (using image segmentation techniques). treadscan.Extractor - creates a tire model from the ellipse, then "unwraps" the tire tread (or rather part of the tire tread, as only about a quarter of the tread is visible).

There are 3 different ways to detect car wheels in image. Only treadscan.Segmentor is included in the treadscan package and works out of the box. This class has a major downfall and that is that it only has a chance to work correctly if the car in the image has very bright rims, which contrast enough with the tire. There is also treadscan.RCNNSegmentor which uses a region based convolutional neural network to find the car wheel in image but the treadscan package doesn't include a pre-trained model. You can however use the model included in this repository or even train your own. You will also need to install torch and torchvision libraries to use the RCNN. The last option is to annotate the image manually using a GUI script, manual_annotation.py, which is included in this folder.

This GUI utilises the OpenCV window, using mouse position and keyboard input to place the points. So to place a point, user can hover over a location and press a specific key, or hold the key while dragging with the mouse.
The points which have to be placed manually are:

• T - top of car wheel (not the tire but the rim)
• B - bottom of car wheel
• R - right (or left) side

These 3 points form an ellipse, which defines the location of car wheel in image (red ellipse in the above image). Then the tire model is extrapolated from that, or can be further modified using the optional points:

• O - move outer tire ellipse (left yellow ellipse)
• S - tire sidewall height (height of yellow ellipses)
• W - tire width (move inner tire ellipse, right yellow ellipse)
• U - top of tread extraction (starting angle, top of green rectangle)
• I - bottom of tread extraction (ending angle, bottom of green rectangle)

Pressing SPACEBAR shows preview of extracted tread, BACKSPACE hides this preview. Pressing ENTER saves full size extracted tread as new image. Closing the window or pressing ESCAPE closes the window.

Additionally the treadscan_script.py in the extras folder can process footage (folder of images/video/stream) and extract the tread of each car from the footage. It has 3 modes: manual (user annotates each manually), semiauto (user annotates only those where Segmentor fails) and auto (images where Segmentor fails are discarded). Extracted treads are saved to a folder.