Use float32 to reduce memory footprint.

[master-thesis.git] / Parasitemia / Parasitemia / Program.fs

diff --git a/Parasitemia/Parasitemia/Program.fs b/Parasitemia/Parasitemia/Program.fs
index 94d2999..09f4312 100644 (file)
--- a/Parasitemia/Parasitemia/Program.fs
+++ b/Parasitemia/Parasitemia/Program.fs
@@ -9,6 +9,9 @@ open System.Windows.Shapes
 open System.Windows.Controls
 open System.Drawing
 open System.Diagnostics
+open System.Threading
+
+open FSharp.Collections.ParallelSeq
 
 open Emgu.CV
 open Emgu.CV.Structure
@@ -63,26 +66,25 @@ let main args =
 
                 initialAreaOpen = 2000
 
-                minRbcRadius = -0.32
-                maxRbcRadius = 0.32
+                minRbcRadius = -0.32f
+                maxRbcRadius = 0.32f
 
                 preFilterSigma = 1.7 // 1.5
 
                 factorNbPick = 1.0
-                factorWindowSize = 2.0
 
-                darkStainLevel = 0.25 // Lower -> more sensitive. 0.3
-                maxDarkStainRatio = 0.1
+                darkStainLevel = 0.22 // Lower -> more sensitive. 0.3. Careful about illumination on the borders.
+                maxDarkStainRatio = 0.1 // 10 %
 
-                infectionArea = 0.012 // 1.2 %
+                infectionArea = 0.012f // 1.2 %
                 infectionLevel = 1.12 // Lower -> more sensitive.
 
-                stainArea = 0.08 // 8 %
+                stainArea = 0.08f // 8 %
                 stainLevel = 1.1 // Lower -> more sensitive.
                 maxStainRatio = 0.12 // 12 %
 
-                standardDeviationMaxRatio = 0.58 // 0.55
-                minimumCellArea = 0.5 })
+                standardDeviationMaxRatio = 0.5 // 0.55
+                minimumCellArea = 0.5f })
 
         match mode with
         | CmdLine (input, output) ->
@@ -102,16 +104,25 @@ let main args =
 
             use resultFile = new StreamWriter(new FileStream(Path.Combine(output, "results.txt"), FileMode.Append, FileAccess.Write))
 
-            for file in files do
-                try
-                    use img = new Image<Bgr, byte>(file)
-                    Utils.log (sprintf "== File: %A" file)
+            //try
+            let images = seq { for file in files -> Path.GetFileNameWithoutExtension(FileInfo(file).Name), new Image<Bgr, byte>(file) }
+
+            let nbConcurrentTaskLimit = 4
+            let n = Environment.ProcessorCount
 
-                    let cells = Utils.logTime "Whole analyze" (fun () -> ImageAnalysis.doAnalysis img (Path.GetFileNameWithoutExtension(FileInfo(file).Name)) config)
+            Utils.logTime "Whole analyze" (fun () ->
+                let results =
+                    images
+                    |> PSeq.map (fun (id, img) -> id, ImageAnalysis.doAnalysis img id (config.Copy()))
+                    |> PSeq.withDegreeOfParallelism (if n > nbConcurrentTaskLimit then nbConcurrentTaskLimit else n)
+
+                for id, cells in results do
                     let total, infected = Utils.countCells cells
-                    fprintf resultFile "File: %s %d %d %.2f\n" file total infected (100. * (float infected) / (float total))
-                with
-                | :? IOException as ex -> Utils.log (sprintf "Unable to open the image '%A': %A" file ex)
+                    fprintf resultFile "File: %s %d %d %.2f\n" id total infected (100. * (float infected) / (float total)))
+
+            //Utils.log (sprintf "== File: %A" file)
+            //with
+            //| :? IOException as ex -> Utils.log (sprintf "Unable to open the image '%A': %A" file ex)
             0
 
         | Window ->