hukaixuan19970627/yolov5_obb这个模型可以在cpu下运行吗？

[123Liyanxi]

目前我的电脑不能运行gpu,之前用cpu运行yolov5成功了，但我需要在遥感数据集中的结果
所以现在在运行https://github.com/hukaixuan19970627/yolov5_obb模型，出现报错：
ImportError: cannot import name 'nms_rotated_ext' from 'utils.nms_rotated' (D:\yolov5_obb-master\utils\nms_rotated_init_.py)
但我看网上很多解决办法是基于gpu训练的解决结果，所以这些方法适用于cpu训练出现的问题吗？

[thalia0503]

你这个看起来应该还没到gpu那步， 这份代码你报错的这个地方应该是你没编译nms_rotated导致的
命令是： python setup.py develop