我是一个中学生,当时我在上关于MMEdu的课,但是我编好代码后,却报错了。
from MMEdu import MMClassification as cls
import os
with open("pid.pkl","w")as f:
f.write(str(os.getpid()))
def generated_train():
model = cls(backbone='LeNet')
model.num_classes = 3
model.load_dataset(path='D:/xedu/XEdu/dataset/cls/hand_gray')
model.save_fold = 'D:/xedu/XEdu/save'
model.train(epochs=5,validate=True,device='cpu',optimizer="SGD",lr=0.001,weight_decay=0.001,checkpoint=None,random_seed=1)
if __name__ == '__main__':
generated_train()
D:\Program Files\Python38\lib\site-packages\mmcv\cnn\bricks\transformer.py:33: UserWarning: Fail to import ``MultiScaleDeformableAttention`` from ``mmcv.ops.multi_scale_deform_attn``, You should install ``mmcv-full`` if you need this module.
warnings.warn('Fail to import ``MultiScaleDeformableAttention`` from '
Traceback (most recent call last):
File "C:\Users\FIX\Desktop\PythonApplication1\PythonApplication1.py", line 1, in
from MMEdu import MMClassification as cls
File "D:\Program Files\Python38\lib\site-packages\MMEdu\__init__.py", line 3, in
from .Detection import MMDetection
File "D:\Program Files\Python38\lib\site-packages\MMEdu\Detection\__init__.py", line 1, in
from .Detection_Edu import MMDetection
File "D:\Program Files\Python38\lib\site-packages\MMEdu\Detection\Detection_Edu.py", line 6, in
from mmdet.apis import inference_detector, init_detector, show_result_pyplot, train_detector
File "D:\Program Files\Python38\lib\site-packages\mmdet\apis\__init__.py", line 2, in
from .inference import (async_inference_detector, inference_detector,
File "D:\Program Files\Python38\lib\site-packages\mmdet\apis\inference.py", line 7, in
from mmcv.ops import RoIPool
File "D:\Program Files\Python38\lib\site-packages\mmcv\ops\__init__.py", line 2, in
from .active_rotated_filter import active_rotated_filter
File "D:\Program Files\Python38\lib\site-packages\mmcv\ops\active_rotated_filter.py", line 8, in
ext_module = ext_loader.load_ext(
File "D:\Program Files\Python38\lib\site-packages\mmcv\utils\ext_loader.py", line 13, in load_ext
ext = importlib.import_module('mmcv.' + name)
File "D:\Program Files\Python38\lib\importlib\__init__.py", line 127, in import_module
return _bootstrap._gcd_import(name[level:], package, level)
ModuleNotFoundError: No module named 'mmcv._ext'
我尝试过更改mmcv安装方式,但在MMEdu安装时,就直接绑定了mmcv-full,无法更改,此办法不可行
代码正常运行,不报错
报错是没有这个模块,安装一下就好了,打开命令行终端:
pip install mmcv-full==1.4.2-f https://download.openmmlab.com/mmcv/dist/cu101/torch1.7.0/index.html