Untitled

 avatar
user_9363972
python
7 months ago
1.1 kB
4
Indexable
Never
import numpy as np
import cv2

min_confidence = 0.5
net = cv2.dnn.readNetFromCaffe(
    "models/deploy.prototxt.txt", "models/res10_300x300_ssd_iter_140000.caffemodel"
)
image = cv2.imread("images/masked.jpg")


height, width = image.shape[0], image.shape[1]

blob = cv2.dnn.blobFromImage(cv2.resize(
    image, (300, 300)), 1.0, (300, 300), (104.0, 117.0, 123.0))
net.setInput(blob)
detections = net.forward()

for i in range(0, detections.shape[2]):
    confidence = detections[0, 0, i, 2]

    if confidence > min_confidence:
        box = detections[0, 0, i, 3:7] * \
            np.array([width, height, width, height])
        (startX, startY, endX, endY) = box.astype('int')
        text = "{:.2f}%".format(confidence * 100)
        y = startY - 10 if startY - 10 > 10 else startY+10
        print(startX, startY, endX, endY)
        cv2.rectangle(image, (startX, startY), (endX, endY), (0, 0, 255), 2)
        cv2.putText(image, text, (startX, y),
                    cv2.FONT_HERSHEY_SIMPLEX, 0.45, (0, 0, 255), 2)

cv2.imshow("Output", image)
cv2.waitKey(0)
cv2.destroyAllWindows()