- Notifications
You must be signed in to change notification settings - Fork54
A flutter plugin for pytorch model inference. Supports image models as well as custom models.
License
NotificationsYou must be signed in to change notification settings
fbelderink/flutter_pytorch_mobile
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
A flutter plugin for pytorch model inference, supported both for Android and iOS.
To use this plugin, addpytorch_mobile as adependency in your pubspec.yaml file.
Create aassets folder with your pytorch model and labels if needed. Modifypubspec.yaml accoringly.
assets: -assets/models/model.pt -assets/labels.csv
Runflutter pub get
import'package:pytorch_mobile/pytorch_mobile.dart';
Either custom model:
Model customModel=awaitPyTorchMobile .loadModel('assets/models/custom_model.pt');
Or image model:
Model imageModel=awaitPyTorchMobile .loadModel('assets/models/resnet18.pt');
List prediction=await customModel .getPrediction([1,2,3,4], [1,2,2],DType.float32);
String prediction=await _imageModel .getImagePrediction(image,224,224,"assets/labels/labels.csv");
final mean= [0.5,0.5,0.5];final std= [0.5,0.5,0.5];String prediction=await _imageModel .getImagePrediction(image,224,224,"assets/labels/labels.csv", mean: mean, std: std);
About
A flutter plugin for pytorch model inference. Supports image models as well as custom models.
Topics
Resources
License
Uh oh!
There was an error while loading.Please reload this page.
Stars
Watchers
Forks
Releases
No releases published
Packages0
No packages published
Uh oh!
There was an error while loading.Please reload this page.
Contributors4
Uh oh!
There was an error while loading.Please reload this page.