java - 为什么某些对象检测神经网络在 OpenCV 4.1.0 中返回全零？

Question

我在OpenCV 4.1.从 Java/Scala 评估 0 中的几个神经网络时遇到问题。网络为鱼自行车图像以及其他图像返回全零。我观察到这一点：

• COCO SSD512* https://github.com/weiliu89/caffe/tree/ssd
• fast_rcnn_inception_v2_coco，ssd_mobilenet_v2_coco https://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/detection_model_zoo.md

检测适用于 YOLOv3-spp 和 YOLOv3-tiny https://pjreddie.com/darknet/yolo/。

这个 dnn 烹饪有什么问题？

// 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))

标签： javascalaopencvtensorflowcaffe