|
@@ -2,6 +2,7 @@ package neuralnet
|
|
|
|
|
|
import (
|
|
|
"errors"
|
|
|
+ "log"
|
|
|
"os/exec"
|
|
|
"path/filepath"
|
|
|
"runtime"
|
|
@@ -105,6 +106,7 @@ func AnalysisPhotoYOLO3(filename string) ([]*ImageClass, error) {
|
|
|
networkConf := "cfg/yolov3.cfg"
|
|
|
if !filesystem.FileExists(filepath.Join(filepath.Dir(darknetBinary), "yolov3.weight")) {
|
|
|
//yolo3 weight not exists. Switch to yolov3_tiny (default, recommended)
|
|
|
+ log.Println("[neuralnet] yolo3.weight not found. Using yolo3-tiny.weight instead")
|
|
|
pretrainWeight = "yolov3-tiny.weights"
|
|
|
networkConf = "cfg/yolov3-tiny.cfg"
|
|
|
}
|