| import io |
| from PIL import Image |
| import cv2 |
| import numpy as np |
| try: |
| from petrel_client.client import Client |
| except ImportError as E: |
| "petrel_client.client cannot be imported" |
| pass |
|
|
|
|
| def pil_loader(img_str): |
| buff = io.BytesIO(img_str) |
| return Image.open(buff) |
|
|
| def cv2_loader(img_bytes): |
| |
| img_mem_view = memoryview(img_bytes) |
| img_array = np.frombuffer(img_mem_view, np.uint8) |
| imgcv2=cv2.imdecode(img_array, cv2.IMREAD_COLOR) |
| imgcv2=cv2.cvtColor(imgcv2, cv2.COLOR_BGR2RGB) |
| return Image.fromarray(imgcv2) |
|
|
| class TCSLoader(object): |
|
|
| def __init__(self, conf_path): |
| self.client = Client(conf_path) |
|
|
| def __call__(self, fn): |
| try: |
| img_value_str = self.client.get(fn) |
| img = pil_loader(img_value_str) |
| except: |
| try: |
| img = cv2_loader(img_value_str) |
| except: |
| print('Read image failed ({})'.format(fn)) |
| return None |
| else: |
| return img |
| else: |
| return img |
|
|
