def test_detect_face(self): orig_image = Image("jpg", "1", "./data/1.jpg") orig_image.getRGB(cache=True) orig_image.detect_face() self.assertTrue(os.path.exists("./data/1_face.jpg"))
def detect_face_from_url(): if request.method == 'POST': url = request.form['url'] filename = url.split('/')[-1] image_file = requests.get(url) if image_file and allowed_file(filename): with open(os.path.join(facerec.config['UPLOAD_FOLDER'], filename), 'wb') as f: f.write(image_file.content) f.close() image = Image('jpg', filename.split(".")[0], os.path.join(facerec.config['UPLOAD_FOLDER'], filename), facerec.config) image.getRGB(cache=True) largestBox = detector.getLargestFaceBoundingBox(image.rgb) alignedFace = detector.alignImg("homography", 256, image.rgb, largestBox, outputPrefix=image.name, outputDebug=True, expandBox=False) #image.detect_face() orig_filename = filename.split('.')[0] + "-orig" + ".jpg" face_imagename = filename.split('.')[0] + "-annotated" + ".jpg" cropped_filename = filename.split('.')[0] + "-cropped" + ".jpg" entries = [] entries.append(orig_filename) entries.append(face_imagename) entries.append(cropped_filename) return render_template('index.html', entries = entries)
def test_save_jpg(self): #Read jpg file to image orig_image = Image("jpg", "1", "./data/1.jpg") orig_image.getRGB(cache=True) # Save it to name + test - jpg image filename = orig_image.name + "_test" orig_image.save('jpg', filename, "./data/") #Check there is file called name + test image in directory self.assertTrue(os.path.exists("./data/1_test.jpg")) os.remove('./data/1_test.jpg')
def detect_face_from_local(filename): image = Image('jpg', filename.split(".")[0], os.path.join(facerec.config['UPLOAD_FOLDER'], filename), facerec.config) image.getRGB(cache=True) largestBox = detector.getLargestFaceBoundingBox(image.rgb) alignedFace = detector.alignImg("homography", 256, image.rgb, largestBox, outputPrefix=image.name, outputDebug=True, expandBox=False) orig_filename = filename.split('.')[0] + "-orig" + ".jpg" face_filename = filename.split('.')[0] + "-annotated" + ".jpg" cropped_filename = filename.split('.')[0] + "-cropped" + ".jpg" entries = [] entries.append(orig_filename) entries.append(face_filename) entries.append(cropped_filename) return render_template('index.html', entries = entries)
def detect_face_from_file(): if request.method == 'POST': image_file = request.files['file'] if image_file and allowed_file(image_file.filename): image_file.save(os.path.join(facerec.config['UPLOAD_FOLDER'], image_file.filename)) image = Image('jpg', image_file.filename.split(".")[0], os.path.join(facerec.config['UPLOAD_FOLDER'], image_file.filename), facerec.config) image.getRGB(cache=True) largestBox = detector.getLargestFaceBoundingBox(image.rgb) alignedFace = detector.alignImg("homography", 256, image.rgb, largestBox, outputPrefix=image.name, outputDebug=True, expandBox=False) #image.detect_face() orig_filename = image_file.filename.split('.')[0] + "-orig" + ".jpg" face_imagename = image_file.filename.split('.')[0] + "-annotated" + ".jpg" cropped_filename = image_file.filename.split('.')[0] + "-cropped" + ".jpg" entries = [] entries.append(orig_filename) entries.append(face_imagename) entries.append(cropped_filename) return render_template('index.html', entries = entries)