ModuleNotFoundError: No module named 'yolc_head'

[JYe9]

您好，我正在尝试使用VisDrone2019数据集来跟着跑一遍，但我在运行 python gen_crop.py 的时候出现了模块缺失的情况:

(yolc) [jye00001 YOLC]$ python gen_crop.py
Traceback (most recent call last):
  File "/data/jye00001/YOLC/gen_crop.py", line 10, in <module>
    from yolc_head import YOLCHead
ModuleNotFoundError: No module named 'yolc_head'

我是根据 get_started.md 进行的环境配置，并且运行环境也满足README里面要求的：

- Python >= 3.8
- PyTorch >= 1.7.0
- mmdetection == 2.26.0 (>=2.17.0, <3.0.0)
- kornia == 0.6.9

[dawn-ech]

改成from models.dense_heads.yolc_head import YOLCHead应该就可以了，感谢指出问题