Использование модели Caffe с OpenCV - PullRequest
Новая
       24

Использование модели Caffe с OpenCV

1 голос
/ 12 июня 2019

Я пытаюсь использовать yahoo nsfw модель с OpenCV. Вот что я попробовал. Я только что скопировал 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 } }
}

Что я сделал не так?

...