YOLOv5 学習 ImportErrorでtrain.pyを実行できない

Question

至急回答頂けると助かります。

https://www.youtube.com/watch?v=GfDTWjiOtBY&t=405s の動画を参考にして、YOLOv5で学習を行ったのですが、
!python train.py --img 416 --batch 16 --epochs 300 --data data.yaml --cfg yolov5n.yaml
を実行したところ
Traceback (most recent call last):
File "train.py", line 46, in <module>
from utils.torch_utils import EarlyStopping, ModelEMA, de_parallel, intersect_dicts, select_device,
ImportError: cannot import name 'intersect_dicts' from 'utils.torch_utils' (/content/drive/My Drive/yolov5/utils/torch_utils.py)

とImportErrorが出力されたのですが、この原因と対処を教えて頂きたいです。

Answer 1

from utils.torch_utils import EarlyStopping, ModelEMA, de_parallel, intersect_dicts, select_device,

を

from utils.torch_utils import EarlyStopping, ModelEMA, de_parallel, select_device,
from utils.general import intersect_dicts

とすれば、そのエラーは消えます。
しかし、utils.generalからもっといろいろなものをimportする必要があるような気がします。

Comments

[klsdja]

ご返答ありがとうございます。せっかく、ご返答頂いたのに申し訳ないですが、学習データのラベリングのミスに気づき、修正するとエラー無く実行できました。お騒がせしてすみません。