import cv2 import numpy as np from PIL import Image, ImageDraw def main(): image = Image.open('D:\\MACH\\MACH\\LAB_2\\img1.jpg') 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()