Function worker

quant.py:215–233

(paths)

    q = Queue()

    def worker(paths):
        def read_img(path, norm=True):
            import cv2
            a = cv2.imread(path)
            a = cv2.resize(a, (224, 224))
            a = a.astype(np.float32)
            a = cv2.cvtColor(a, cv2.COLOR_BGR2RGB)
            if norm:
                a /= 255
                # mean and std for RGB images
                a -= [0.485, 0.456, 0.406]
                a /= [0.229, 0.224, 0.225]
            a = np.moveaxis(a, -1, 0)
            return a

        for i in range(0, len(paths), batch_size):
            xs = np.stack(list(map(lambda x: read_img(x, True), paths[i:i + batch_size])))
            q.put(xs)
        q.put(None)

    image_exts = ['JPEG', 'jpg', 'jpeg', 'png']
    filenames = list(itertools.chain(*[glob.glob(os.path.join(image_dir, '**/*.' + ext), recursive=True)