henry99a commited on
Commit
847a12b
·
1 Parent(s): ca7217f

Fix startup panic when cascade data is missing

Browse files
Files changed (1) hide show
  1. detector/detector.go +6 -1
detector/detector.go CHANGED
@@ -24,10 +24,15 @@ const (
24
  var classifier *pigo.Pigo
25
 
26
  func init() {
27
- classifier, _ = pigo.NewPigo().Unpack(cascade)
 
 
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}