python - ImportError：没有名为“generate_xml”的模块

Question

我正在尝试使用 yolo 模型创建一个关于实时车辆分类的项目。当我尝试注释车辆图像时，会弹出此错误：

Traceback (most recent call last):

  File "train.py", line 5, in <module>

    from generate_xml import write_xml

ImportError: No module named 'generate_xml'

我用来训练的来源可以在这里找到。

Answer 1

只需重命名part7 - generate_xml.py并generate_xml.py确保它位于运行脚本的同一位置