From a1eca9b3526593d29b03300000e3b597a62e37b3 Mon Sep 17 00:00:00 2001 From: jwansek Date: Tue, 25 Aug 2020 03:27:52 +0100 Subject: added facial recognision --- get_images.py | 30 +++++++++++++++++++++++++----- 1 file changed, 25 insertions(+), 5 deletions(-) (limited to 'get_images.py') diff --git a/get_images.py b/get_images.py index 08a17e3..b4fef24 100644 --- a/get_images.py +++ b/get_images.py @@ -83,14 +83,34 @@ class DownloadedImage: def __exit__(self, type, value, traceback): os.remove(self.filename) -if __name__ == "__main__": - # print(get_image(get_random_searchtag())) +def main(): + try: + simg = get_image(get_random_searchtag()) + except ConnectionError: + main() - simg = get_image(get_random_searchtag()) with DownloadedImage(simg.imurl) as impath: img = cv2.imread(impath) - cv2.imshow("img, ", img) - cv2.waitkey(0) + cascade = cascade = cv2.CascadeClassifier(os.path.join("lbpcascade_animeface", "lbpcascade_animeface.xml")) + gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY) + gray = cv2.equalizeHist(gray) + + faces = cascade.detectMultiScale(gray, + # detector options + scaleFactor = 1.1, + minNeighbors = 5, + minSize = (24, 24)) + for (x, y, w, h) in faces: + cv2.rectangle(img, (x, y), (x + w, y + h), (0, 0, 255), 2) + + cv2.imshow("Press to exit", img) + cv2.waitKey(0) + +if __name__ == "__main__": + main() + + + -- cgit v1.2.3