yolox: AssertionError: Caught AssertionError in DataLoader worker process 0.

YOLOX训练报错解决：PNG图片格式不匹配问题

最新推荐文章于 2024-12-22 23:29:16 发布

原创

最新推荐文章于 2024-12-22 23:29:16 发布 · 1w 阅读

4 ·

CC 4.0 BY-SA版权

文章标签：

#pytorch

在使用YOLOX进行目标检测训练时遇到错误，问题源于YOLOX源码预设读取JPEG格式图片，而实际数据集为PNG格式。在VOC.py的图像读取部分，将代码中jpg格式修改为png后，成功解决了读取异常。此问题提醒我们在使用预定义模型时需确保数据格式与模型要求一致。

yolox 训练报错如下代码块：

根据错误提示，在voc.py文件中读取图片时发生异常。
原因：yolox给的源码读取的是lpg格式的图片，而我的输入是png格式的图片，所以报错。
（也有可能是路径没有给正确，无法读取到图片数据）

AssertionError: Caught AssertionError in DataLoader worker process 0.
Original Traceback (most recent call last):
  File "/home/crush/ProgramFiles/anaconda3/lib/python3.7/site-packages/torch/utils/data/_utils/worker.py", line 287, in _worker_loop
    data = fetcher.fetch(index)
  File "/home/crush/ProgramFiles/anaconda3/lib/python3.7/site-packages/torch/utils/data/_utils/fetch.py", line 44, in fetch
    data = [self.dataset[idx] for idx in possibly_batched_index]
  File "/home/crush/ProgramFiles/anaconda3/lib/python3.7/site-packages/torch/utils/data/_utils/fetch.py", line 44, in <listcomp>
    data = [self.dataset[idx] for idx in possibly_batched_index]
  File "/home/crush/YOLOX/yolox/data/datasets/datasets_wrapper.py", line 110, in wrapper
    ret_val = getitem_fn(self, index)
  File "/home/crush/YOLOX/yolox/data/datasets/mosaicde