【yolov5】yolov5训练模型出现错误

训练自己数据集的时候需要在yaml文件下正确填写训练集及验证集的路径。

使用下列语句执行训练时出现以下错误：

python train.py --batch-size 2 --epoch 200 --data data/leaf/data.yaml --weights .\yolov5s.pt

Initializing libiomp5md.dll, but found libiomp5md.dll already initialized.

这个问题通常是由于多个线程同时调用了OpenMP库（例如libiomp5md.dll）而导致的。通常来说，可以通过设置环境变量KMP_DUPLICATE_LIB_OK为TRUE来解决这个问题，这样就可以允许库的重复加载，从而避免了这种问题。

在train.py文件中插入以下代码即可解决：

import os
os.environ['KMP_DUPLICATE_LIB_OK'] = 'True'