class LoadImages: def __init__(self, path): if '*' in path: files = sorted(glob.glob(path, recursive=True)) # glob elif os.path.isdir(path): files = sorted(glob.glob(os.path.join(path, '*.*'))) # dir elif os.path.isfile(path): files = [path] # files else: raise Exception(f'ERROR: {path} does not exist')
加载数据集,加载文件夹下所有文件
最新推荐文章于 2022-10-21 11:21:08 发布