Revision history [back]

(Object Detection)prior_box_layer.cpp error using Opencv dnn

I using Ubuntu 16.04.3, built Opencv 3.4.0 from source using this for python 3.5. image description

I try to use Wiki method in myself training data.

I use tf_text_graph_ssd.py script to generate a text graph representation.

On running the sample provided as object-detection-opencv.py

import cv2 as cv

cvNet = cv.dnn.readNetFromTensorflow('frozen_inference_graph.pb', 'out_graph.pbtxt')

img = cv.imread('image2.jpg')
rows = img.shape[0]
cols = img.shape[1]
cvNet.setInput(cv.dnn.blobFromImage(img, 1.0/127.5, (300, 300), (127.5, 127.5, 127.5), swapRB=True, crop=False))
cvOut = cvNet.forward()

for detection in cvOut[0,0,:,:]:
    score = float(detection[2])
    if score > 0.3:
        left = detection[3] * cols
        top = detection[4] * rows
        right = detection[5] * cols
        bottom = detection[6] * rows
        cv.rectangle(img, (int(left), int(top)), (int(right), int(bottom)), (23, 230, 210), thickness=2)

cv.imshow('img', img)
cv.waitKey()

Result: image description

shared here:https://drive.google.com/open?id=1Npu54EGHtqdd6dFb2Ff3TxRb5UpE1xkP

Any fix for this would be very much helpful.