Fix startup panic when cascade data is missing
Browse files- detector/detector.go +6 -1
detector/detector.go
CHANGED
|
@@ -24,10 +24,15 @@ const (
|
|
| 24 |
var classifier *pigo.Pigo
|
| 25 |
|
| 26 |
func init() {
|
| 27 |
-
|
|
|
|
|
|
|
| 28 |
}
|
| 29 |
|
| 30 |
func detectFaces(params *pigo.CascadeParams, angles ...float64) []pigo.Detection {
|
|
|
|
|
|
|
|
|
|
| 31 |
// initialize angles if empty.
|
| 32 |
if len(angles) == 0 {
|
| 33 |
angles = []float64{0.0}
|
|
|
|
| 24 |
var classifier *pigo.Pigo
|
| 25 |
|
| 26 |
func init() {
|
| 27 |
+
if len(cascade) > 0 {
|
| 28 |
+
classifier, _ = pigo.NewPigo().Unpack(cascade)
|
| 29 |
+
}
|
| 30 |
}
|
| 31 |
|
| 32 |
func detectFaces(params *pigo.CascadeParams, angles ...float64) []pigo.Detection {
|
| 33 |
+
if classifier == nil {
|
| 34 |
+
return nil
|
| 35 |
+
}
|
| 36 |
// initialize angles if empty.
|
| 37 |
if len(angles) == 0 {
|
| 38 |
angles = []float64{0.0}
|