6917d8a3dbf6617f02b1247fd50fbc20d3cb6fd4,codes/data/util.py,,get_image_paths,#,21

Before Change

    images = []
    for dirpath, _, fnames in sorted(os.walk(path)):
        for fname in sorted(fnames):
            if is_image_file(fname):
                img_path = os.path.join(dirpath, fname)
                images.append(img_path)
    assert images, "%s has no valid image file" % path
    return images

After Change

def get_image_paths(data_type, dataroot):
    env, paths = None, None
    if dataroot is not None:
        if data_type == "lmdb":
            env, paths = _get_paths_from_lmdb(dataroot)
        elif data_type == "img":
            paths = sorted(_get_paths_from_images(dataroot))
        else:
            raise NotImplementedError("data_type [%s] is not recognized." % data_type)
    return env, paths


def _read_lmdb_img(env, path):

Instances