def find_people_in_frame(inputFrame, detector=HOGDescriptor_getDefaultPeopleDetector):
hog = HOGDescriptor()
hog.setSVMDetector(detector())
image = resize(inputFrame, width=min(400, inputFrame.shape[1]))
orig = image.copy()
(rects, _) = hog.detectMultiScale(
image, winStride=(4, 4), padding=(8, 8), scale=1.05)
for (x, y, w, h) in rects:
rectangle(orig, (x, y), (x+w, y+h), (255, 0, 0), 2)
rects = array([[x, y, x+w, y+h] for (x, y, w, h) in rects])
merged_rects = nms(rects, probs=None, overlapThresh=0.65)
for (xA, yA, xB, yB) in merged_rects:
rectangle(image, (xA, yA), (xB, yB), (0, 255, 0), 2)
size_of_image = image.shape[0] * image.shape[1]
size_of_window = abs(xA - xB) * abs(yA - yB)
portion_occupied = (size_of_image / size_of_window) * 100
color = None
if portion_occupied >= 0.1:
color = (255, 0, 0)
elif portion_occupied >= 0.05:
color = (0, 0, 255)
else:
color = (0, 255, 0)
putText(
image, f'Threat Level: {round(abs(xA - xB) * abs(yA - yB) * 1 / 150)}', (xA, yA-10), FONT_HERSHEY_SIMPLEX, 0.9, color, 2)
return image
def run_detection_baseline(inputFolder, outputFolder):
hog = HOGDescriptor()
hog.setSVMDetector(HOGDescriptor_getDefaultPeopleDetector())
if not exists(outputFolder):
mkdir(outputFolder)
for imagePath in list_images(inputFolder):
image = imread(imagePath)
image = resize(image, width=min(400, image.shape[1]))
orig = image.copy()
(rects, _) = hog.detectMultiScale(
image, winStride=(4, 4), padding=(8, 8), scale=1.05)
for (x, y, w, h) in rects:
rectangle(orig, (x, y), (x+w, y+h), (0, 0, 255), 2)
rects = array([[x, y, x+w, y+h] for (x, y, w, h) in rects])
merged_rects = nms(rects, probs=None, overlapThresh=0.65)
for (xA, yA, xB, yB) in merged_rects:
rectangle(image, (xA, yA), (xB, yB), (0, 255, 0), 2)
filename = imagePath[imagePath.rfind('/') + 1:]
imwrite(join(outputFolder, filename), image)