summaryrefslogtreecommitdiff
path: root/megapixels/commands/cv/face_roi.py
diff options
context:
space:
mode:
authorAdam Harvey <adam@ahprojects.com>2019-01-07 18:49:09 +0100
committerAdam Harvey <adam@ahprojects.com>2019-01-07 18:49:09 +0100
commitc67a3c23190287f747989e5dc7725e1657edf8f1 (patch)
tree5ff47ab72e6fb146f66730db06fc3b91265dd769 /megapixels/commands/cv/face_roi.py
parent55b9734d131a197166156566d1b999a8bb59169b (diff)
add age, gender, emotion stubs
Diffstat (limited to 'megapixels/commands/cv/face_roi.py')
-rw-r--r--megapixels/commands/cv/face_roi.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/megapixels/commands/cv/face_roi.py b/megapixels/commands/cv/face_roi.py
index 6d42924e..70fff401 100644
--- a/megapixels/commands/cv/face_roi.py
+++ b/megapixels/commands/cv/face_roi.py
@@ -94,11 +94,11 @@ def cli(ctx, opt_fp_in, opt_dir_media, opt_fp_out, opt_data_store, opt_dataset,
if opt_detector_type == types.FaceDetectNet.CVDNN:
detector = face_detector.DetectorCVDNN()
elif opt_detector_type == types.FaceDetectNet.DLIB_CNN:
- detector = face_detector.DetectorDLIBCNN(opt_gpu)
+ detector = face_detector.DetectorDLIBCNN(gpu=opt_gpu)
elif opt_detector_type == types.FaceDetectNet.DLIB_HOG:
detector = face_detector.DetectorDLIBHOG()
elif opt_detector_type == types.FaceDetectNet.MTCNN:
- detector = face_detector.DetectorMTCNN()
+ detector = face_detector.DetectorMTCNN(gpu=opt_gpu)
elif opt_detector_type == types.FaceDetectNet.HAAR:
log.error('{} not yet implemented'.format(opt_detector_type.name))
return