Merge pull request #63 from GDSC-DEU/feat/#62-update-latest-model

최신 모델 파일로 업데이트
GDSC-DEU · Jun 5, 2022 · 4313c4d · 4313c4d
2 parents f67b442 + 00be959
commit 4313c4d
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/client/assets/tflite/candrink-v4-640-f32-large.tflite b/client/assets/tflite/candrink-v4-640-f32-large.tflite
diff --git a/client/assets/tflite/converted_model-640-fp16-small-v1.tflite b/client/assets/tflite/converted_model-640-fp16-small-v1.tflite
diff --git a/client/lib/services/tflite/classifier.dart b/client/lib/services/tflite/classifier.dart
@@ -8,12 +8,12 @@ import 'package:tflite_flutter/tflite_flutter.dart';
 import 'package:tflite_flutter_helper/tflite_flutter_helper.dart';
 
 class Classifier {
-  static const MODEL_FILENAME = 'tflite/converted_model-640-fp16-small-v1.tflite';
+  static const MODEL_FILENAME = 'tflite/candrink-v4-640-f32-large.tflite';
   static const LABELS_FILENAME = 'assets/tflite/labels.txt';
 
   /// Input size of image (height = width = 640)
   static const INPUT_SIZE = 640;
-  static const THRESHOLD = 0.8;
+  static const THRESHOLD = 0.5;
 
   // Number of results to show
   static const int NUM_RESULTS = 10;
@@ -45,7 +45,7 @@ class Classifier {
     this.interpreter = interpreter ??
         await Interpreter.fromAsset(
           MODEL_FILENAME,
-          options: InterpreterOptions()..threads = 4,
+          options: InterpreterOptions()..threads = 8,
         );
 
     _outputShapes = this.interpreter!.getOutputTensors().map((tensor) => tensor.shape).toList();

diff --git a/client/lib/ui/camera_view.dart b/client/lib/ui/camera_view.dart
@@ -68,7 +68,7 @@ class _CameraViewState extends State<CameraView> with WidgetsBindingObserver {
   initCamera() async {
     cameras = await availableCameras();
 
-    cameraController = CameraController(cameras[0], ResolutionPreset.low, enableAudio: false);
+    cameraController = CameraController(cameras[0], ResolutionPreset.ultraHigh, enableAudio: false);
     await cameraController!.initialize().then((_) async {
       await cameraController!.startImageStream(onLatestImageAvailable);