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

使用windows进行YOLO8的自定义数据训练时报错ForkingPickler(file, protocol).dump(obj)BrokenPipeError: [Errno 32] Broken pipe

一. 问题出现如下：

二. 错误原因：

该问题的产生是由于windows下多线程的问题，和DataLoader类有关。将num_workers的数量修改为0即可。感觉这是windows下的一个bug。

三. 解决办法：

一般情况下，更改train.py里的 num_workers=1 改为0 就可，如下：

trainset = torchvision.datasets.CIFAR10(root='./data',train=True,
                                          download=True,transform=transform)
trainloader = torch.utils.data.DataLoader(trainset,batch_size=4,
                                            shuffle = True,num_workers=1) #改为0

但是我找了半天也没找到yolo8的train.py的num_workers在哪里，最后发现将build.py中的build_dataloader函数返回的num_workers更改了就行，如下：

def build_dataloader(cfg, batch, img_path, stride=32, rect=False, names=None, rank=-1, mode='train'):...

文件的具体位置在：

你安装的虚拟环境\Lib\site-packages\ultralytics\yolo\data\build.py

亲测有效！