Почему некоторые нейронные сети обнаружения объектов возвращают все нули в OpenCV 4.1.0? - PullRequest
1 голос
/ 04 июля 2019

У меня проблема с оценкой нескольких нейронных сетей в OpenCV 4.1.0 из Java / Scala. Сети возвращают все нули для изображения fish-bike , а также другие. Я наблюдаю это в:

Обнаружение работает с YOLOv3-spp и YOLOv3-tiny https://pjreddie.com/darknet/yolo/.

Что может быть не так в этой кухне?

    // The reproduce in Scala REPL you need a hack:

    def loadLibraryScalaREPL(libraryName: String): Unit = {
      val loadLibrary0 = Runtime.getRuntime.getClass.getDeclaredMethods()(4)
      loadLibrary0.setAccessible(true)
      loadLibrary0.invoke(Runtime.getRuntime, scala.tools.nsc.interpreter.IMain.getClass, libraryName)
    }

    loadLibraryScalaREPL(org.opencv.core.Core.NATIVE_LIBRARY_NAME)

    // To load in Java/Scala application just use System.loadLibrary.

    import org.opencv.core.{Scalar, Size}
    import org.opencv.dnn.Dnn
    import org.opencv.imgcodecs.Imgcodecs

    val image = Imgcodecs.imread("/root/fish-bike.jpg")
    val net = Dnn.readNetFromCaffe("/tmp/VGG_coco_SSD_512x512_iter_360000.prototxt", "/tmp/VGG_coco_SSD_512x512_iter_360000.caffemodel")
    val blob = Dnn.blobFromImage(image, 1/125.0, new Size(512, 512), new Scalar(104,117,123), true)
    net.setInput(blob)
    val layer = net.forward()
    val values = new Array[Float](layer.total().toInt)
    layer.get(0,0, values)
    values.grouped(7).foreach(x => println(x.toList))
...