解释 images = set() infer_dir = os.path.abspath(infer_dir) assert os.path.isdir(infer_dir), \ "infer_dir {} is not a directory".format(infer_dir) exts = ['jpg', 'jpeg', 'png', 'bmp'] exts += [ext.upper() for ext in exts] for ext in exts: images.update(glob.glob('{}/*.{}'.format(infer_dir, ext))) images = list(images) assert len(images) > 0, "no image found in {}".format(infer_dir) logger.info("Found {} inference images in total.".format(len(images))) return images
时间: 2024-03-29 09:40:52 浏览: 75
RSeQC.infer_experiment:RSeQC.infer_GenePattern的实验包装器
这段代码的作用是在指定目录中查找所有的图片文件,并将其存储在一个列表中返回。具体步骤如下:
1. 定义一个空的集合 images。
2. 将 infer_dir 转换为绝对路径,并确保其是一个目录,如果不是则抛出异常。
3. 定义一个包含常见图片文件扩展名的列表 exts。
4. 遍历 exts 列表中的所有扩展名,使用 glob.glob 函数查找所有该扩展名的文件,并将其添加到 images 集合中。
5. 将 images 集合转换为列表并返回,如果列表为空则抛出异常。
6. 打印日志信息,指出在指定目录中找到的图片数量。
这段代码的主要目的是用于读取图片文件,以便进行 AI 推理等操作。
阅读全文