X-Git-Url: http://git.euphorik.ch/?p=master-thesis.git;a=blobdiff_plain;f=Parasitemia%2FParasitemia%2FMatchingEllipses.fs;h=92a767ad19d14907c890fc23315852853eb24af6;hp=8a1b4f8e89ffc26139e7024bc2cba1baad59ca32;hb=b070295cf67b2025164a34b6594e84f0d771cdc9;hpb=e76da913cd58078ad2479357b2430ed62a6e0777

diff --git a/Parasitemia/Parasitemia/MatchingEllipses.fs b/Parasitemia/Parasitemia/MatchingEllipses.fs
index 8a1b4f8..92a767a 100644
--- a/Parasitemia/Parasitemia/MatchingEllipses.fs
+++ b/Parasitemia/Parasitemia/MatchingEllipses.fs
@@ -9,33 +9,43 @@ open Types
 open Utils
 
 
+// Do not take in account matching score below this when two ellipses are matched.
 let matchingScoreThreshold1 = 0.6
-let matchingScoreThreshold2 = 1.0
+
+// All ellipsee with a score below this is removed.
+let matchingScoreThreshold2 = 1. / 50.
 
 type private EllipseScoreFlaggedKd (matchingScore: float, e: Ellipse) =
     let mutable matchingScore = matchingScore
+    let perimeter = e.Perimeter
 
     member this.Ellipse = e
 
     member this.MatchingScore = matchingScore
+
+    // The score is proportional to the perimeter because large ellipse will receive more votes.
     member this.AddMatchingScore(score: float) =
-        matchingScore <- matchingScore + score
+        matchingScore <- matchingScore + score / perimeter
 
     member val Processed = false with get, set
     member val Removed = false with get, set
 
-    interface KdTree.I2DCoords with    
+    interface KdTree.I2DCoords with
         member this.X = this.Ellipse.Cx
         member this.Y = this.Ellipse.Cy
 
-        
+
 type MatchingEllipses (radiusMin: float) =
     let ellipses = List<EllipseScoreFlaggedKd>()
-        
-    member this.Add (e: Ellipse) = 
+
+    member this.Add (e: Ellipse) =
         ellipses.Add(EllipseScoreFlaggedKd(0.0, e))
 
     member this.Ellipses : Ellipse list =
+        List.ofSeq ellipses |> List.map (fun e -> e.Ellipse)
+
+    // Process all ellipses and return ellipses ordered from the best score to the worst.
+    member this.PrunedEllipses : Ellipse list =
         if ellipses.Count = 0
         then
             []
@@ -43,13 +53,13 @@ type MatchingEllipses (radiusMin: float) =
             // 1) Create a kd-tree from the ellipses list.
             let tree = KdTree.Tree.BuildTree (List.ofSeq ellipses)
 
-            // 2) Compute the matching score of each ellipses.       
+            // 2) Compute the matching score of each ellipses.
             let windowSize = radiusMin
             for e in ellipses do
                 e.Processed <- true
                 let areaE = e.Ellipse.Area
                 let window = { KdTree.minX = e.Ellipse.Cx - windowSize / 2.0
-                               KdTree.maxX = e.Ellipse.Cx + windowSize / 2.0 
+                               KdTree.maxX = e.Ellipse.Cx + windowSize / 2.0
                                KdTree.minY = e.Ellipse.Cy - windowSize / 2.0
                                KdTree.maxY = e.Ellipse.Cy + windowSize / 2.0 }
                 for other in tree.Search window do
@@ -68,7 +78,7 @@ type MatchingEllipses (radiusMin: float) =
             // 3) Sort ellipses by their score.
             ellipses.Sort(fun e1 e2 -> e2.MatchingScore.CompareTo(e1.MatchingScore))
 
-            // 4) Remove ellipses wich have a low score. 
+            // 4) Remove ellipses wich have a low score.
             let i = ellipses.BinarySearch(EllipseScoreFlaggedKd(matchingScoreThreshold2, Ellipse(0.0, 0.0, 0.0, 0.0, 0.0)),
                                           { new IComparer<EllipseScoreFlaggedKd> with
                                                 member this.Compare(e1, e2) = e2.MatchingScore.CompareTo(e1.MatchingScore) }) |> abs
@@ -84,18 +94,18 @@ type MatchingEllipses (radiusMin: float) =
                 if not e.Removed
                 then
                     let window = { KdTree.minX = e.Ellipse.Cx - e.Ellipse.A
-                                   KdTree.maxX = e.Ellipse.Cx + e.Ellipse.A 
+                                   KdTree.maxX = e.Ellipse.Cx + e.Ellipse.A
                                    KdTree.minY = e.Ellipse.Cy - e.Ellipse.A
                                    KdTree.maxY = e.Ellipse.Cy + e.Ellipse.A }
                     for other in tree.Search window do
                         if not other.Removed && other.MatchingScore < e.MatchingScore
-                        then            
-                            if e.Ellipse.Contains other.Ellipse.Cx other.Ellipse.Cy
+                        then
+                            if e.Ellipse.Scale(0.8).Contains other.Ellipse.Cx other.Ellipse.Cy
                             then
                                 other.Removed <- true
             ellipses.RemoveAll(fun e -> e.Removed) |> ignore
 
             List.ofSeq ellipses |> List.map (fun e -> e.Ellipse)
 
-        
+