torch_mobile

Description:

torch mobile

torch_mobile #
A Flutter plugin for inference of Pytorch models.
Note: This plugin is still under development, only image classification models are supported for the moment.
Installation #
First, add torch_mobile as a dependency in your pubspec.yaml file.
iOS #
Not implemented yet
Android #
No configuration required - the plugin should work out of the box.
Usage #
Create a assets folder with pytorch model and labels file and model file in it.
Modify pubspec.yaml accordingly.
assets:
- assets/model.pt
- assets/labels.txt
copied to clipboard
Import the library
import 'package:torch_mobile/torch_mobile.dart';
copied to clipboard
Load model and labels
TorchMobile.loadModel(model: 'assets/model.pt', labels: 'assets/labels.txt');
copied to clipboard
Get prediction for image
String prediction = await TorchMobile.getPrediction(image, maxWidth: 400, maxHeight: 400);
copied to clipboard