package/pigo
1
0
Fork 0
mirror of https://github.com/esimov/pigo.git synced 2025-12-24 12:48:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: #46 new method to avoid running into an infinite loop

Browse Source
This commit is contained in:
esimov
2021-01-22 12:26:25 +02:00
parent 294912e8d8
commit e9f068f1a3
2 changed files with 6 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
core/pigo.go
View File

@@ -243,10 +243,11 @@ func (pg *Pigo) RunCascade(cp CascadeParams, angle float64) []Detection {
}
}
}
// 100 is an arbitrary value to speedup the process.
// Normally we should have been used 1/scale, but this would been icreased the iteration process.
// This approach will result in a significant improvement without having an impact on the detection score.
scale = int(float64(scale) + math.Ceil(100/float64(scale)*cp.ScaleFactor))
// We need to avoid running into an infinite loop because of float to int conversion
// in cases when scaleFactor = 1.1 and minSize = 9 as example.
// When the scale is 9, the factor would come up with 9.9, which again becomes 9 because of the int() conversion.
// This approach gives the same speed without having an impact on the detection score.
scale = int(float64(scale) + math.Max(2, (float64(scale)*cp.ScaleFactor)-float64(scale)))
}
return detections
}

2
wasm/index.html
View File

@@ -26,7 +26,7 @@
<body>
<script type="text/javascript">
var stats = new Stats();
stats.showPanel(1);
stats.showPanel(0);
document.body.appendChild(stats.dom);
</script>
</body>