22 lines
555 B
Python
22 lines
555 B
Python
import cv2
|
|
import numpy as np
|
|
from PIL import Image, ImageDraw
|
|
|
|
|
|
def main():
|
|
image = Image.open('input.jpeg')
|
|
image_arr = np.asarray(image)
|
|
|
|
hog = cv2.HOGDescriptor()
|
|
hog.setSVMDetector(cv2.HOGDescriptor_getDefaultPeopleDetector())
|
|
detections, weights = hog.detectMultiScale(image_arr)
|
|
detections_rectangles = detections.tolist()
|
|
draw = ImageDraw.Draw(image)
|
|
for x, y, w, h in detections_rectangles:
|
|
draw.rectangle((x, y, x + w, y + h), outline=(255, 0, 0))
|
|
image.show()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|