Pytorch/Yolov5目标检测报错总结

1. 主要原因是虚拟内存不足,虚拟机的话增加虚拟机内存即可
2. win10系统下,电脑内存分配是优先程序的,可以在"高级系统设置"–>“高级”–>“性能设置”–>"高级"里修改电脑内存
   

3. 基于yolov5,发现Win10系统下不允许python进行多线程操作的,那么只需要在train.py文件内将dataloader的num_workers参数修改为0即可

#将原始train.py文件中的dataloader中workers参数改为0
    dataloader, dataset = create_dataloader(train_path, imgsz, batch_size, gs, opt,
                                            hyp=hyp, augment=True, cache=opt.cache_images, rect=opt.rect, rank=rank,
                                            world_size=opt.world_size, workers=0,
                                            image_weights=opt.image_weights)

---

  parser.add_argument('--batch-size', type=int, default=4, help='total batch size for all GPUs')

---

配置的nc不正确,检查一下数据集的分类,将yaml里的nc改为分类大小即可