[BUG] inference.py 期望 --data-dir 目录下包含带标签的子文件夹
作者: JosuaRieder创建于 2024年11月4日更新于 2026年8月1日
标签bug
** 描述错误** 当运行'inform.py-data-dir data-class-map class map.txt-model efficiencynet b7-num-class 8 - checkpoint 输出/model best.pth.tar'时,“data”是一个包含图像且没有子文件夹的目录,脚本会与以下输出相撞:
回溯( 最近最后一次通话) :
文件 "/home/josua/redacted/pytorch-image-models/inference.py", 第369行,以<module>
主要( )
file "/home/josua/redacted/pytorch-image-models/ inference.py", 第224行,主要内容为:
数据集=创建 数据集 ()
^^^^^^^^^^^^^^^
文件"/home/josua/redacted/pytorch-image-models/tim/data/dataset factory.py",行216,在creat dataset中
ds = 图像数据集( I)
^^^^^^^^^^^^^
文件 "/home/josua/redacted/pytorch-image-models/tim/data/dataset.py", 第36行,收于 init
阅读器 = 创建 reader ()
^^^^^^^^^^^^^^
文件"/home/josua/redacted/pytorch-image-models/tim/data/readers/reader factory.py",第44行,创建-reader
读取器 = 读取器 ImageFolder(根,**kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
文件"/home/josua/redacted/pytorch-image-models/timm/data/readers/reader image folder.py",82行,收于 init .
提升运行时错误( R)
运行时错误:在/home/josua/redacted/data/unlabed的子文件夹中发现了0个图像. 支持的图像扩展为:.png,.jpg,.jpeg** 预期行为** 我期望它不提出错误,而是对`数据 ' 的(反向)内容进行推论,而不论是否有标签子文件夹,因为标签对于推论没有必要。
** Desktop(请填写下列资料):**
- OS:x86 64 Linux 6.11.5-arch1-1
- 这个寄存器版本: d4dde48d
- PyTorch版本 w/ CUDA/cuDNN:Python-pytorch-cuda 2.3.1-8,cuda 12.6.2- 2个
** 其他情况** 工作周:在不直接将图像放入“数据/”而是将其放入“数据/随机 标签/”时,“随机-标签”是一个随机标签,即来自“class map.txt”的随机行
内容来源: huggingface/pytorch-image-models