Commit 2505f1bd authored by dmMaze's avatar dmMaze
Browse files

support read webp

parent c9a0bdbf

ballontranslator/utils/io_utils.py

+1 −1
Original line number Diff line number Diff line
@@ -3,7 +3,7 @@ import os.path as osp 
import numpy as np

from pathlib import Path



IMG_EXT = ['.bmp', '.jpg', '.png', '.jpeg']

IMG_EXT = ['.bmp', '.jpg', '.png', '.jpeg', '.webp']

NP_BOOL_TYPES = (np.bool_, np.bool8)

NP_FLOAT_TYPES = (np.float_, np.float16, np.float32, np.float64)

NP_INT_TYPES = (np.int_, np.int8, np.int16, np.int32, np.int64, np.uint, np.uint8, np.uint16, np.uint32, np.uint64)