Goto

Collaborating Authors

 pytorch mobile neural network


Object Detector Android App Using PyTorch Mobile Neural Network

#artificialintelligence

Running Machine Learning code on mobile devices is the next big thing. PyTorch, in the latest release PyTorch 1.3, added PyTorch Mobile for deploying machine learning models on Android and iOS devices. Here we will look into creating an Android Application for object detection inside an image; like the GIF shown below. We will be using a pre-trained ResNet18 model for this tutorial. ResNet18 is the state of the art computer vision model with 1000 classes for classification.