pytorch_lite 3.0.0+1 pytorch_lite: ^3.0.0+1 copied to clipboard
Flutter package to help run pytorch lite models classification and yolov5 and yolov8
import 'package:flutter/material.dart';
import 'package:pytorch_lite/image_utils_isolate.dart';
import 'package:pytorch_lite/native_wrapper.dart';
import 'package:pytorch_lite/pytorch_lite.dart';
import 'package:pytorch_lite_example/run_model_by_camera_demo.dart';
import 'package:pytorch_lite_example/run_model_by_image_demo.dart';
void main() {
PytorchFfi.init();
ImageUtilsIsolate.init();
runApp(const ChooseDemo());
}
class ChooseDemo extends StatefulWidget {
const ChooseDemo({Key? key}) : super(key: key);
@override
State<ChooseDemo> createState() => _ChooseDemoState();
}
class _ChooseDemoState extends State<ChooseDemo> {
@override
Widget build(BuildContext context) {
return MaterialApp(
home: Scaffold(
appBar: AppBar(
title: const Text('Pytorch Mobile Example'),
),
body: Builder(builder: (context) {
return Center(
child: Column(
children: [
TextButton(
onPressed: () => {
Navigator.push(
context,
MaterialPageRoute(
builder: (context) => const RunModelByCameraDemo()),
)
},
style: TextButton.styleFrom(
backgroundColor: Colors.blue,
),
child: const Text(
"Run Model with Camera",
style: TextStyle(
color: Colors.white,
),
),
),
TextButton(
onPressed: () => {
Navigator.push(
context,
MaterialPageRoute(
builder: (context) => const RunModelByImageDemo()),
)
},
style: TextButton.styleFrom(
backgroundColor: Colors.blue,
),
child: const Text(
"Run Model with Image",
style: TextStyle(
color: Colors.white,
),
),
)
],
),
);
}),
),
);
}
}