diff --git a/kaldi_data_generator.py b/kaldi_data_generator.py index 7271768e64f0fe4413290b75615fb403b8bd325c..dbe0a73222a66c74eb2564d9121aba052dd130db 100644 --- a/kaldi_data_generator.py +++ b/kaldi_data_generator.py @@ -10,11 +10,10 @@ from io import BytesIO from pathlib import Path from typing import Tuple -import tqdm - import cv2 import numpy as np import requests +import tqdm from PIL import Image from apistar.exceptions import ErrorResponse from arkindex import ArkindexClient, options_from_env @@ -47,8 +46,8 @@ def download_image(url): # Preprocess the image and prepare it for classification image = Image.open(BytesIO(resp.content)) logger.debug('Downloaded image {} - size={}x{}'.format(url, - image.size[0], - image.size[1])) + image.size[0], + image.size[1])) return image @@ -117,7 +116,7 @@ class KaldiDataGenerator: full_image_url = res['zone']['image']['s3_url'] if full_image_url is None: - full_image_url = res['zone']['image']['url'] + '/full/full/0/default.jpg' + full_image_url = res['zone']['image']['url'] + '/full/full/0/default.jpg' img = self.get_image(full_image_url, page_id=page_id)