open System
open System.IO
-open System.Windows
-open System.Windows.Media
-open System.Windows.Markup
-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
-open Emgu.CV.WPF
open Config
-let display (window : Views.MainWindow) (img : IImage) =
- let imgControl = window.Root.FindName("img") :?> Controls.Image
- imgControl.Source <- BitmapSourceConvert.ToBitmapSource(img)
-
-let log (window : Views.MainWindow) (mess : string) =
- let txtLog = window.Root.FindName("txtLog") :?> Controls.TextBlock
- txtLog.Text <- txtLog.Text + mess + "\n"
-
-
type Input =
| File of string
| Dir of string
[<EntryPoint>]
+[<STAThread()>]
let main args =
match parseArgs args with
| mode, debug ->
let config =
Config(
- { scale = 1.
+ {
+ initialAreaOpen = 2000
- minRbcRadius = -0.32
- maxRbcRadius = 0.32
+ minRbcRadius = -0.3f
+ maxRbcRadius = 0.3f
- preFilterSigma = 1.5
+ preFilterSigma = 1.7 // 1.5
factorNbPick = 1.0
- factorWindowSize = 2.0
- darkStainLevel = 0.3 // Lower -> more sensitive.
- maxDarkStainRatio = 0.1
+ darkStainLevel = 0.25 // Lower -> more sensitive. 0.3. Careful about illumination on the borders.
+ maxDarkStainRatio = 0.1 // 10 %
- infectionArea = 0.012 // 1.2 %
- infectionLevel = 0.2
- parasitePixelsRequired = 1
+ infectionArea = 0.012f // 1.2 %
+ infectionLevel = 1.12 // Lower -> more sensitive.
- stainArea = 0.08
- stainLevel = 0.1
+ stainArea = 0.08f // 8 %
+ stainLevel = 1.1 // Lower -> more sensitive.
maxStainRatio = 0.12 // 12 %
- standardDeviationMaxRatio = 0.55
- minimumCellArea = 0.5 })
+ standardDeviationMaxRatio = 0.5 // 0.55
+ minimumCellArea = 0.5f })
match mode with
| CmdLine (input, output) ->
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 ->
- let app = new Application()
- let mainWindow = Views.MainWindow()
-
- if debug
- then
- config.Debug <- DebugOn "."
+ (*let display (window : Views.MainWindow) (img : IImage) =
+ let imgControl = window.Root.FindName("img") :?> Controls.Image
+ imgControl.Source <- BitmapSourceConvert.ToBitmapSource(img)
- Utils.log <- (fun m -> log mainWindow m)
+ let log (window : Views.MainWindow) (mess : string) =
+ let txtLog = window.Root.FindName("txtLog") :?> Controls.TextBlock
+ txtLog.Text <- txtLog.Text + mess + "\n"*)
- //display mainWindow img
- mainWindow.Root.Show()
- app.Run()
+ if debug then config.Debug <- DebugOn "."
+ GUI.run config