使用 Caffe、OpenCV 和 C# 进行 CNN 分割

Question

我在 C# 有一个项目，我需要在其中使用图像分割。 我主要从pyimagesearch.com了解了 DNN 和 OpenCV（感谢 Adrian，很棒的网站。）但是：我不明白我应该如何从结果中提取单个片段。 这是python中的部分代码：

blob = cv2.dnn.blobFromImage(frame, 1 / 255.0, (1024, 512), 0, swapRB=True, crop=False)
net.setInput(blob)
start = time.time()
output = net.forward()
end = time.time()
# infer the total number of classes along with the spatial
# dimensions of the mask image via the shape of the output array
(numClasses, height, width) = output.shape[1:4]
# our output class ID map will be num_classes x height x width in
# size, so we take the argmax to find the class label with the
# largest probability for each and every (x, y)-coordinate in the
# image
classMap = np.argmax(output[0], axis=0)
# given the class ID map, we can map each of the class IDs to its
# corresponding color
mask = COLORS[classMap]
# resize the mask such that its dimensions match the original size
# of the input frame
mask = cv2.resize(mask, (frame.shape[1], frame.shape[0]), interpolation=cv2.INTER_NEAREST)
# perform a weighted combination of the input frame with the mask
# to form an output visualization
output = ((0.3 * frame) + (0.7 * mask)).astype("uint8")

我能够创建网络，加载.caffemodel，创建blob并运行“forward（）”但我不知道如何编写等效的

(numClasses, height, width) = output.shape[1:4]
classMap = np.argmax(output[0], axis=0)
mask = COLORS[classMap]

在 C# 中（我正在为 OpenCV 使用 EMGU 包装器）

这是我的代码：

Emgu.CV.Dnn.Net segNet = Emgu.CV.Dnn.DnnInvoke.ReadNetFromCaffe("segmentation_deploy.prototxt", "fcn8s-heavy-pascal.caffemodel");
Mat rawImage = Emgu.CV.CvInvoke.Imread(_hideImagePath, ImreadModes.AnyColor);
double scale = 1/255;
Mat blob = Emgu.CV.Dnn.DnnInvoke.BlobFromImage(rawImage, scale, rawImage.Size, new MCvScalar(104.00698793, 116.66876762, 122.67891434), true, false);
segNet.SetInput(blob);
Mat results = segNet.Forward();
Mat tmpMat = results.Reshape(0, results.SizeOfDimension[2]);
ImageViewer vTL1 = new ImageViewer(tmpMat, "tmpMat");
vTL1.Show();

它仅显示宽黑色矩形。 提前感谢您的帮助。