From c67a3c23190287f747989e5dc7725e1657edf8f1 Mon Sep 17 00:00:00 2001 From: Adam Harvey Date: Mon, 7 Jan 2019 18:49:09 +0100 Subject: add age, gender, emotion stubs --- megapixels/app/processors/face_beauty.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'megapixels/app/processors/face_beauty.py') diff --git a/megapixels/app/processors/face_beauty.py b/megapixels/app/processors/face_beauty.py index 2e8221b7..a01c6834 100644 --- a/megapixels/app/processors/face_beauty.py +++ b/megapixels/app/processors/face_beauty.py @@ -9,12 +9,10 @@ import numpy as np import imutils import pickle -os.environ['CUDA_VISIBLE_DEVICES'] = '' import keras from keras.layers import Conv2D, Input, MaxPool2D,Flatten, Dense, Permute, GlobalAveragePooling2D from keras.models import Model from keras.optimizers import adam -import os.path from keras.models import Sequential from keras.applications.resnet50 import ResNet50 #from keras.applications.resnet50 import Dense @@ -33,7 +31,9 @@ class FaceBeauty: # Estimates beauty using CNN - def __init__(self, gpu=-1): + def __init__(self): + # ensure Keras uses CPU (no GPU, it takes up all memory) + os.environ['CUDA_VISIBLE_DEVICES'] = '' # don't really need GPU, CPU is quick enough self.log = logger_utils.Logger.getLogger() resnet = ResNet50(include_top=False, pooling='avg') -- cgit v1.2.3-70-g09d2