在自定义数据集上训练YOLOv8检测模型遇启动进程错误求助
解决YOLOv8训练时的RuntimeError问题
问题原因
Windows系统中,Python多进程启动要求主模块代码必须包裹在if __name__ == '__main__':代码块内,否则会触发进程初始化错误。YOLOv8训练过程会启动子进程执行任务,你的代码未遵循这个规范,因此报错。
修改后的代码
将训练逻辑放入主模块入口判断中:
from ultralytics import YOLO if __name__ == '__main__': model = YOLO("yolov8n.yaml") results = model.train(data=r"C:\Users\abdal\PycharmProjects\bus_car_detection\venv\config.yaml", epochs=20)
额外检查项
- 确认
config.yaml路径格式:Windows下可改用正斜杠/或双反斜杠\\避免转义问题,比如把train: train\images改为train: train/images或train: train\\images - 确保
E:\dataset目录下存在train/images、valid/images文件夹,且对应标注文件(默认同目录下的labels文件夹)完整
内容的提问来源于stack exchange,提问作者Abdalrhman Mogahed
相关产品推荐
相关产品推荐

