Ask Your Question
0

How can I use a custom tensorflow model with the CV2 DNN module?

asked 2018-07-11 11:16:27 -0600

jarulsamy gravatar image

I retrained an object detection model based on Google's Tensorflow object detection API. I exported it as a frozen inference graph. I would like to use it with CV2's DNN module:

cap = cv2.VideoCapture(URL)

cvNet = cv2.dnn.readNetFromTensorflow('graph.pb', 'graph.pbtxt')

while True:
    ret, img = cap.read()
    rows = img.shape[0]
    cols = img.shape[1]
    cvNet.setInput(cv2.dnn.blobFromImage(img, 1.0/127.5, (300, 300), (127.5, 127.5, 127.5), swapRB=True, crop=False))
    cvOut = cv2Net.forward()

    for detection in cv2Out[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
            cv2.rectangle(img, (int(left), int(top)), (int(right), int(bottom)), (23, 230, 210), thickness=2)
    cv2.imshow('img', img)
    if cv2.waitKey(1) ==27:
        exit(0)

I get this error: Const input blob for weights not found in function getConstBlob

From my research, I believe I have to optimize the inference graph. I can't find any documentation as how to do this.

If anyone could point me in the right direction, it would be very much appreciated.

edit retag flag offensive close merge delete

Comments

1

it's only preliminary, but please have a look here and also here

berak gravatar imageberak ( 2018-07-11 11:20:48 -0600 )edit

@berak I tried the troubleshooting and am having the same issue.

jarulsamy gravatar imagejarulsamy ( 2018-07-11 14:44:14 -0600 )edit

1 answer

Sort by ยป oldest newest most voted
0

answered 2020-05-23 23:58:09 -0600

frame = vs.read() frame = imutils.resize(frame, width=400)

# grab the frame dimensions and convert it to a blob
(h, w) = frame.shape[:2]
blob = cv2.dnn.blobFromImage(cv2.resize(frame, (300, 300)),
    0.007843, (300, 300), 127.5)
edit flag offensive delete link more

Question Tools

1 follower

Stats

Asked: 2018-07-11 11:16:27 -0600

Seen: 3,182 times

Last updated: Jul 11 '18