diff --git a/deploy/mtcnn_detector.py b/deploy/mtcnn_detector.py index 2044f1d..c7332a5 100644 --- a/deploy/mtcnn_detector.py +++ b/deploy/mtcnn_detector.py @@ -6,7 +6,11 @@ import math import cv2 from multiprocessing import Pool from itertools import repeat -from itertools import izip +try: + from itertools import izip +except ImportError: + izip = zip + from helper import nms, adjust_input, generate_bbox, detect_first_stage_warpper class MtcnnDetector(object):