io_object.py 文件源码

def load_rgb(data_path):
    im_path = os.path.join(data_path,'image_2','*.png')
    im_files = sorted(glob.glob(im_path))

    im_all = []

    #tic = time.time()
    for iter_files in im_files:            
        if len(im_all)<100 :
            im = np.uint8(mpimg.imread(iter_files) * 255)
            im_all.append(im) 
            print(im.shape)
    return im_all