在OpenCV中使用Caffe模型
在OpenCV中使用Caffe模型
提问于 2019-06-12 06:30:40
我正在尝试在OpenCV中使用yahoo nsfw模型。这是我尝试过的。我刚刚从存储库中复制了deploy.prototxt和resnet_50_1by2_nsfw.caffemodel。
import cv2 as cv
cvNet = cv.dnn.readNetFromCaffe('deploy.prototxt','resnet_50_1by2_nsfw.caffemodel')
image_path = 'adult.jpg'
image = cv.imread(image_path)
blob = cv.dnn.blobFromImage(image, 1, (224,224), (0,0,0), True, crop=False)
cvNet.setInput(blob)
detections = cvNet.forward()
print(detections)无论我有什么图像作为输入，我得到的输出都是这样的[[0.9855554 0.01444463]]，第一个值总是更大。我之所以选择图像大小(224, 224)，是因为在deploy.prototxt中，我发现了以下内容：
name: "ResNet_50_1by2_nsfw"
layer {
name: "data"
type: "Input"
top: "data"
input_param { shape: { dim: 1 dim: 3 dim: 224 dim: 224 } }
}我做错什么了？
回答 1
Stack Overflow用户
发布于 2019-10-08 11:44:49
我不确定你是否已经解决了这个问题，我也不是这方面的专家，但我发现你在创建blob时不执行均值减法(在你的代码中均值被设置为(0,0,0) )，这是非常奇怪的。查看您链接的Github存储库，作者确实在https://github.com/yahoo/open_nsfw/blob/master/classify_nsfw.py第114行执行了均值减法：
caffe_transformer.set_mean('data', np.array([104, 117, 123]))这可能会对网络的分类产生影响。我建议在OpenCV实现中使用它们的参数。
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/56555903
复制
![Matlab画图线型、符号及颜色汇总[通俗易懂]](https://ask.qcloudimg.com/http-save/yehe-8223537/dbefa1a8a6cde0cf7e2df41a918bf8a4.png)
相似问题
围绕物体旋转
围绕旋转物体旋转物体？
围绕中心物体旋转物体
围绕着不断旋转的物体旋转物体
three.js如何围绕相机坐标旋转物体？
添加站长 进交流群
领取专属 10元无门槛券
AI混元助手 在线答疑
关注 腾讯云开发者公众号
洞察 腾讯核心技术
剖析业界实践案例
腾讯云开发者
