6 open FSharp.Collections.ParallelSeq
16 let doAnalysis (img
: Image<Bgr, byte
>) (name
: string) (config
: Config) (reportProgress
: (int -> unit) option) : Cell list =
17 // To report the progress of this function from 0 to 100.
18 let inline report
(percent
: int) =
19 match reportProgress
with
23 use green = img
.Item(1)
24 let greenFloat = green.Convert<Gray, float32
>()
25 let filteredGreen = gaussianFilter
greenFloat config.LPFStandardDeviation
27 let initialAreaOpening = int<| config.RBCArea * config.Parameters.ratioAreaPaleCenter
28 logTime
"Area opening number one" (fun () -> ImgTools.areaOpenF
filteredGreen initialAreaOpening)
32 let delta = config.Parameters.granulometryRange
* config.RBCRadius
33 int <| config.RBCRadius - delta, int <| config.RBCRadius + delta
34 //let r1 = logTime "Granulometry (morpho)" (fun() -> Granulometry.findRadiusByClosing (filteredGreen.Convert<Gray, byte>()) range 1.0 |> float32)
35 let r2 = logTime
"Granulometry (area)" (fun() -> Granulometry.findRadiusByAreaClosing
filteredGreen range |> float32
)
36 // log (sprintf "r1: %A, r2: %A" r1 r2)
37 config.RBCRadius <- r2
40 let secondAreaOpening = int <| config.RBCArea * config.Parameters.ratioAreaPaleCenter
41 if secondAreaOpening > initialAreaOpening
43 logTime
"Area opening number two" (fun () -> ImgTools.areaOpenF
filteredGreen secondAreaOpening)
45 let parasites, filteredGreenWhitoutStain
= ParasitesMarker.find
filteredGreen config
46 //let parasites, filteredGreenWhitoutInfection, filteredGreenWhitoutStain = ParasitesMarker.findMa greenFloat filteredGreenFloat config
48 let edges, xGradient
, yGradient
= logTime
"Finding edges" (fun () -> ImgTools.findEdges
filteredGreenWhitoutStain)
49 logTime
"Removing small connected components from thinning" (fun () -> removeArea
edges (config.RBCRadius ** 2.f
/ 50.f
|> int))
51 let allEllipses, ellipses
= logTime
"Finding ellipses" (fun () ->
52 let matchingEllipses = Ellipse.find
edges xGradient yGradient
config
53 matchingEllipses.Ellipses, matchingEllipses.PrunedEllipses)
56 let cells = logTime
"Classifier" (fun () -> Classifier.findCells
ellipses parasites filteredGreenWhitoutStain config)
59 // Output pictures if debug flag is set.
60 match config.Debug with
62 let dirPath = System.IO.Path.Combine(output
, name
)
63 System.IO.Directory.CreateDirectory dirPath |> ignore
65 let buildFileName postfix
= System.IO.Path.Combine(dirPath, name
+ postfix
)
67 saveMat
(edges * 255.0) (buildFileName " - edges.png")
69 saveImg
parasites.darkStain
(buildFileName " - parasites - dark stain.png")
70 saveImg
parasites.stain
(buildFileName " - parasites - stain.png")
71 saveImg
parasites.infection
(buildFileName " - parasites - infection.png")
73 let imgAllEllipses = img
.Copy()
74 drawEllipses
imgAllEllipses allEllipses (Bgr(0.0, 240.0, 240.0)) 0.05
75 saveImg
imgAllEllipses (buildFileName " - ellipses - all.png")
77 let imgEllipses = filteredGreenWhitoutStain.Convert<Bgr, byte
>()
78 drawEllipses
imgEllipses ellipses (Bgr(0.0, 240.0, 240.0)) 1.0
79 saveImg
imgEllipses (buildFileName " - ellipses.png")
81 let imgCells = img
.Copy()
82 drawCells
imgCells false cells
83 saveImg
imgCells (buildFileName " - cells.png")
85 let imgCells' = img.Copy()
86 drawCells imgCells' true cells
87 saveImg
imgCells' (buildFileName " - cells - full.png")
89 let filteredGreenMaxima = gaussianFilter greenFloat config.LPFStandardDeviation
90 for m in ImgTools.findMaxima filteredGreenMaxima do
91 ImgTools.drawPoints filteredGreenMaxima m 255.f
92 saveImg filteredGreenMaxima (buildFileName " - filtered - maxima.png")
94 saveImg filteredGreen (buildFileName " - filtered.png")
95 saveImg filteredGreenWhitoutStain (buildFileName " - filtered closed stain.png")
96 //saveImg filteredGreenWhitoutInfection (buildFileName " - filtered closed infection.png")
98 saveImg green (buildFileName " - green.png")
100 use blue = img.Item(0)
101 saveImg blue (buildFileName " - blue.png")
103 use red = img.Item(2)
104 saveImg red (buildFileName " - red.png")
109 // ID * cell radius * cell list.
110 let doMultipleAnalysis (imgs: (string * Image<Bgr, byte>) list) (config : Config) (reportProgress: (int -> unit) option) : (string * float * Cell list) list =
111 let inline report (percent: int) =
112 match reportProgress with
113 | Some f -> f percent
116 let monitor = Object()
117 let mutable total = 0
118 let nbImgs = List.length imgs
119 let reportProgressImg =
121 lock monitor (fun () -> total <- total + progress)
122 report (total / nbImgs))
124 let nbConcurrentTaskLimit = 4
125 let n = Environment.ProcessorCount
128 |> PSeq.map (fun (id, img) ->
129 let localConfig = config.Copy()
130 let cells = doAnalysis img id localConfig (Some reportProgressImg)
131 id, float localConfig.RBCRadius, cells)
132 |> PSeq.withDegreeOfParallelism (if n > nbConcurrentTaskLimit then nbConcurrentTaskLimit else n)