This workbook explores the idea of generting heat maps from a pretrained model to create heat maps that allow for rough detection of object in an image. An example of dog detection using this is given below.
- python 3.6
- tensorflow 2.3
- matplotlib 3.3.3
- PIL 8.0.1
- numpy 1.19
- cv2 4.4.0.46
The idea can be taken much further and used to generate bounding boxes. This was done by making a threshold map from heatmaps and then finding coordinates of the object of interest. Please refer to notebook 3 for some examples.
This resulted in the following bounding box
The examples below are some of the tests carried out on other animals.
