tinyML Research Symposium -Short Papers: Classification of Depth-Map Image on AI Microcontroller

https://www.tinyml.org/event/research-symposium-2023/#schedule
Classification of Depth-Map Image on AI Microcontroller
Jesse SANTOS, Software Engineer, Analog Devices
Depth maps are known to help enhancing the visual tasks. In this paper, we consider image classification using depth maps only. The classification task is done on an edge device called MAX78000 artificial intelligence (AI) accelerator. We trained a SimpleNet deep learning model using the generated depth-map images from a publicly available point-cloud dataset. The training was quantization-aware, targeted for MAX78000. The achieved F1-score for 10 classes was 60 to 96 percent. We consider to test the trained model on a dataset that is unknown to it. Thus, we captured depth map images using a Time-of-Flight (ToF) camera to create a dataset for inference on the AI microcontroller.

source