123Liyanxi

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

1

目前我的电脑不能运行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训练出现的问题吗？