Faster inference

trikko · May 12, 2024 · 641e850 · 641e850
1 parent a9c4ecd
commit 641e850
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 3 deletions.
diff --git a/source/ai.d b/source/ai.d
@@ -250,11 +250,19 @@ struct AI
       assert(hasAI);
 
       import mir.algorithm.iteration : minIndex, maxIndex;
+      import picture : Picture;
 
       assert(session !is null);
 
-      Image img = imread(file);
-      scope(exit) destroyFree(img);
+      Image img = new Image(
+         Picture.width,
+         Picture.height,
+         ImageFormat.IF_RGB,
+         BitDepth.BD_8,
+         cast(ubyte[])Picture.pixbuf.getPixelsWithLength()
+      );
+
+      scope(exit) destroy(img);
 
       Slice!(ubyte*, 3) imSlice = img.sliced; // will be freed with the previous destroyFree(img)
 

diff --git a/source/gui.d b/source/gui.d
@@ -474,7 +474,7 @@ struct GUI
 					auto airects = AI.boxes(Picture.current);
 					Picture.rects ~= airects;
 					Picture.historyCommit();
-					confirmAnnotation();
+					Picture.writeAnnotations();
 					canvas.queueDraw();
 				}