Some small modifications

qureai · Jul 28, 2016 · e6eb3a9 · e6eb3a9
1 parent ec83a2f
commit e6eb3a9
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/constants.lua b/constants.lua
@@ -11,6 +11,6 @@ imgHeight = 112 -- Height of the image to be resized to for use
 imgWidth = 144 -- Width of the image to be resized to for use
 trueHeight = 420 -- True height of the image
 trueWidth = 580 -- True width of the image
-baseSegmentationProb = 0.70 -- Probability with which to consider a pixel, a part of mask
+baseSegmentationProb = 0.85 -- Probability with which to consider a pixel, a part of mask
 nbClasses = 2 -- Number of classes in output, 2 since mask/no mask
 interpolation = 'bicubic' -- Interpolation to be used for resizing
diff --git a/utils/utils.lua b/utils/utils.lua
@@ -99,7 +99,6 @@ function CalculateDiceScore(outputs, targets)
     local dice_coeff = 0
     for i=1,outputs:size(1) do
         local output_flatten = GetMaskFromOutput(outputs[i])
-		-- local output_flatten = GetTunedResult(outputs[i],baseSegmentationProb)
         local target_flatten = targets[i]:float():add(-1)
         local numerator = torch.cmul(output_flatten, target_flatten)
         if output_flatten:sum() + target_flatten:sum() ~= 0 then