Skip to content

ImportError: cannot import name 'YOLO11n' from 'yolo11n_eval' (AIV-795) #245

@ceoifung

Description

@ceoifung

Checklist

  • Checked the issue tracker for similar issues to ensure this is not a duplicate
  • Read the documentation to confirm the issue is not addressed there and your configuration is set correctly
  • Tested with the latest version to ensure the issue hasn't been fixed

How often does this bug occurs?

always

Expected behavior

我参考了教程https://docs.espressif.com/projects/esp-dl/zh_CN/latest/tutorials/how_to_quantize_model.html 搭建好了esp-ppq的相关环境。然后根据https://github.com/espressif/esp-dl/blob/578ca7ada61dadc2ff769c46a3e7b689cc37e3ab/examples/tutorial/how_to_quantize_model/quantize_yolo11n/trainer.py 想训练yolov11n的模型,然后量化模型为espdl。

Actual behavior (suspected bug)

实际上执行程序无法执行,报如下错误

root@3dd5e519213c:/workspace/espdl-yolo-train# python trainer.py

    ___________ ____        ____  ____  ____
   / ____/ ___// __ \      / __ \/ __ \/ __ \
  / __/  \__ \/ /_/ /_____/ /_/ / /_/ / / / /
 / /___ ___/ / ____/_____/ ____/ ____/ /_/ /
/_____//____/_/         /_/   /_/    \___\_\


WARNING ⚠️ user config directory '/root/.config/Ultralytics' is not writeable, defaulting to '/tmp' or CWD.Alternatively you can define a YOLO_CONFIG_DIR environment variable for this path.
Traceback (most recent call last):
  File "/workspace/espdl-yolo-train/trainer.py", line 20, in <module>
    from yolo11n_eval import QuantDetectionValidator, YOLO11n
ImportError: cannot import name 'YOLO11n' from 'yolo11n_eval' 

yolo11n_eval.py中不存在YOLO11n

Error logs or terminal output

root@3dd5e519213c:/workspace/espdl-yolo-train# python trainer.py

    ___________ ____        ____  ____  ____
   / ____/ ___// __ \      / __ \/ __ \/ __ \
  / __/  \__ \/ /_/ /_____/ /_/ / /_/ / / / /
 / /___ ___/ / ____/_____/ ____/ ____/ /_/ /
/_____//____/_/         /_/   /_/    \___\_\


WARNING ⚠️ user config directory '/root/.config/Ultralytics' is not writeable, defaulting to '/tmp' or CWD.Alternatively you can define a YOLO_CONFIG_DIR environment variable for this path.
Traceback (most recent call last):
  File "/workspace/espdl-yolo-train/trainer.py", line 20, in <module>
    from yolo11n_eval import QuantDetectionValidator, YOLO11n
ImportError: cannot import name 'YOLO11n' from 'yolo11n_eval' (/workspace/espdl-yolo-train/yolo11n_eval.py)

Steps to reproduce the behavior

  1. 下载quantize_yolo11n的代码
  2. docker部署esp-ppq环境
  3. 运行quantize_yolo11n的trainer.py
  4. 报错,没有YOLO11n

Project release version

master

System architecture

Intel/AMD 64-bit (modern PC, older Mac)

Operating system

Linux

Operating system version

Ubuntu 22.04.4 LTS

Shell

Bash

Additional context

No response

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions