def _load_image(image_file, load_cv2=False): """Loads image using utils (PIL Image) or CV2 imdecode.""" if not load_cv2: return utils.load_image(image_file) # Load images via cv2: with tf.io.gfile.GFile(image_file, 'rb') as f: img_bgr = cv2.imdecode(np.fromstring(f.read(), dtype=np.uint8), cv2.IMREAD_COLOR) # OpenCV reads in BGR format return cv2.cvtColor(img_bgr, cv2.COLOR_BGR2RGB) # Return RGB converted image
def imread(filename): with tf.gfile.Open(filename, 'rb') as f: raw_image = np.asarray(bytearray(f.read()), dtype=np.uint8) image = cv2.imdecode(raw_image, cv2.IMREAD_COLOR) image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB) return image.astype(np.float32) / 255.0