我该如何解决这个YoloV8问题

0 投票
1 回答
44 浏览
提问于 2025-04-14 15:28

有人能告诉我为什么我会遇到这个错误吗?

Traceback (most recent call last):
  File "<string>", line 1, in <module>
  File "D:\Programming\Pythoncode\Resources\lib\multiprocessing\spawn.py", line 116, in spawn_main
    exitcode = _main(fd, parent_sentinel)
  File "D:\Programming\Pythoncode\Resources\lib\multiprocessing\spawn.py", line 125, in _main
    prepare(preparation_data)
  File "D:\Programming\Pythoncode\Resources\lib\multiprocessing\spawn.py", line 236, in prepare
    _fixup_main_from_path(data['init_main_from_path'])
  File "D:\Programming\Pythoncode\Resources\lib\multiprocessing\spawn.py", line 287, in _fixup_main_from_path
    main_content = runpy.run_path(main_path,
  File "D:\Programming\Pythoncode\Resources\lib\runpy.py", line 288, in run_path        
    return _run_module_code(code, init_globals, run_name,
  File "D:\Programming\Pythoncode\Resources\lib\runpy.py", line 97, in _run_module_code 
    _run_code(code, mod_globals, init_globals,
  File "D:\Programming\Pythoncode\Resources\lib\runpy.py", line 87, in _run_code        
    exec(code, run_globals)
  File "d:\Programming\Pythoncode\Computer_Vision\test.py", line 7, in <module>
    results = model.train(data="D:\Programming\Pythoncode\Computer_Vision\data.yaml", epochs=1, imgsz=640)
  File "D:\Programming\Pythoncode\Resources\lib\site-packages\ultralytics\engine\model.py", line 644, in train
    self.trainer.train()
  File "D:\Programming\Pythoncode\Resources\lib\site-packages\ultralytics\engine\trainer.py", line 208, in train
    self._do_train(world_size)
  File "D:\Programming\Pythoncode\Resources\lib\site-packages\ultralytics\engine\trainer.py", line 322, in _do_train
    self._setup_train(world_size)
  File "D:\Programming\Pythoncode\Resources\lib\site-packages\ultralytics\engine\trainer.py", line 286, in _setup_train
    self.train_loader = self.get_dataloader(self.trainset, batch_size=batch_size, rank=RANK, mode="train")
  File "D:\Programming\Pythoncode\Resources\lib\site-packages\ultralytics\models\yolo\detect\train.py", line 55, in get_dataloader
    return build_dataloader(dataset, batch_size, workers, shuffle, rank)  # return dataloader
  File "D:\Programming\Pythoncode\Resources\lib\site-packages\ultralytics\data\build.py", line 114, in build_dataloader
    return InfiniteDataLoader(
  File "D:\Programming\Pythoncode\Resources\lib\site-packages\ultralytics\data\build.py", line 40, in __init__
    self.iterator = super().__iter__()
  File "D:\Programming\Pythoncode\Resources\lib\site-packages\torch\utils\data\dataloader.py", line 439, in __iter__
    return self._get_iterator()
  File "D:\Programming\Pythoncode\Resources\lib\site-packages\torch\utils\data\dataloader.py", line 387, in _get_iterator
    return _MultiProcessingDataLoaderIter(self)
  File "D:\Programming\Pythoncode\Resources\lib\site-packages\torch\utils\data\dataloader.py", line 1040, in __init__
    w.start()
  File "D:\Programming\Pythoncode\Resources\lib\multiprocessing\process.py", line 121, in start
    self._popen = self._Popen(self)
  File "D:\Programming\Pythoncode\Resources\lib\multiprocessing\context.py", line 224, in _Popen
    return _default_context.get_context().Process._Popen(process_obj)
  File "D:\Programming\Pythoncode\Resources\lib\multiprocessing\context.py", line 327, in _Popen
    return Popen(process_obj)
  File "D:\Programming\Pythoncode\Resources\lib\multiprocessing\popen_spawn_win32.py", line 45, in __init__
    prep_data = spawn.get_preparation_data(process_obj._name)
  File "D:\Programming\Pythoncode\Resources\lib\multiprocessing\spawn.py", line 154, in get_preparation_data
    _check_not_importing_main()
  File "D:\Programming\Pythoncode\Resources\lib\multiprocessing\spawn.py", line 134, in _check_not_importing_main
    raise RuntimeError('''
RuntimeError:
        An attempt has been made to start a new process before the
        current process has finished its bootstrapping phase.
This probably means that you are not using fork to start your
child processes and you have forgotten to use the proper idiom
in the main module:

    if __name__ == '__main__':
        freeze_support()
        ...

The "freeze_support()" line can be omitted if the program
is not going to be frozen to produce an executable.

这是主要的代码:

from ultralytics import YOLO



model = YOLO('yolov8n.pt')

results = model.train(data="D:\Programming\Pythoncode\Computer_Vision\data.yaml", epochs=1, imgsz=640)

而data.yaml文件里有这些内容:

train: 
    D:\Programming\Pythoncode\Computer_Vision\train\images val:
    D:\Programming\Pythoncode\Computer_Vision\valid\images test:
    D:\Programming\Pythoncode\Computer_Vision\test\images

nc: 1
names: ['fire']

1 个回答

0

你可以试着用命令行界面来运行,而不是用自定义脚本。

如果不行的话,确保你的 num_workers 设置为 0。我在我的脚本中也遇到过同样的问题,当我把它设置为 1 时不行,改成 0 后就好了。

撰写回答