yolov5 dataloader出现KeyError: ‘__getstate__‘问题

yolov5 dataloader出现KeyError: '__getstate__'问题

问题描述

我从github下载了一个叫TarDal的项目，它是将图像融合然后用yolov5做目标检测的，但是我在运行代码的时候出现了如下错误

    return Popen(process_obj)
  File "D:\anaconda\envs\tardal\lib\multiprocessing\popen_spawn_win32.py", line 93, in __init__
    reduction.dump(process_obj, to_child)
  File "D:\anaconda\envs\tardal\lib\multiprocessing\reduction.py", line 60, in dump
    ForkingPickler(file, protocol).dump(obj)
KeyError: '__getstate__'

解决方法

我搜索关键词ForkingPickler(file, protocol).dump(obj)网络上出现的都是yolo的问题，于是我猜测就是yolov5的问题

将dataloader里面的num_workers改成0就可以运行了

参考文章

使用windows进行YOLO8训练时报错ForkingPickler(file, protocol).dump(obj)BrokenPipeError: [Errno 32] BrokenPipe