|
from PIL import Image
|
|
import scipy.io
|
|
import numpy as np
|
|
|
|
def load_semseg(filename, loader_type):
|
|
if loader_type == 'PIL':
|
|
semseg = np.array(Image.open(filename), dtype=np.int)
|
|
elif loader_type == 'MAT':
|
|
semseg = scipy.io.loadmat(filename)['LabelMap']
|
|
return semseg |