Last active
March 4, 2019 21:54
-
-
Save navarasu/8b004d1897019d3f7833508075d6aa6d to your computer and use it in GitHub Desktop.
Main Dart for Loading TF Lite assets example
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
import 'dart:async'; | |
import 'dart:convert'; | |
import 'package:flutter/material.dart'; | |
import 'package:flutter/services.dart'; | |
import 'package:camera/camera.dart'; | |
List<CameraDescription> cameras; | |
var labels, colors; | |
Future<void> main() async { | |
try { | |
cameras = await availableCameras(); | |
} on CameraException catch (e) { | |
print('Error: $e.code\nError Message: $e.message'); | |
} | |
await _loadModel(); | |
runApp(ObjectDetectApp()); | |
} | |
Future<Null> _loadModel() async { | |
try { | |
const platform = const MethodChannel('francium.tech/tensorflow'); | |
var jsonString = await rootBundle.loadString('assets/yolov2-tiny.meta'); | |
var metaData = json.decode(jsonString); | |
labels = metaData["labels"]; | |
colors = metaData["colors"]; | |
metaData["blockSize"] = 32; | |
metaData["threshold"] = 0.5; | |
metaData["overlap_threshold"] = 0.7; | |
metaData["max_result"] = 15; | |
final String result = await platform.invokeMethod('loadModel', | |
{"modal_path": "assets/yolov2_graph.lite", "meta_data": metaData}); | |
print(result); | |
} on PlatformException catch (e) { | |
print('Error: $e.code\nError Message: $e.message'); | |
} | |
} | |
class ObjectDetectApp extends StatelessWidget { | |
@override | |
Widget build(BuildContext context) { | |
return new MaterialApp( | |
home: Scaffold( | |
appBar: AppBar( | |
title: Center( | |
child: Text('Object Detector'), | |
)), | |
)); | |
} | |
} |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment