(op) D:\picture\conan\mmrotate>python demo/image_demo.py demo/demo.jpg oriented_rcnn_r50_fpn_1x_dota_le90.py oriented_rcnn_r50_fpn_1x_dota_le90-6d2b2ce0.pth --out-file result.jpg
D:\Anaconda3\Anaconda\envs\op\lib\site-packages\mmcv\__init__.py:20: UserWarning: On January 1, 2023, MMCV will release v2.
0.0, in which it will remove components related to the training process and add a data transformation module. In addition,
it will rename the package names mmcv to mmcv-lite and mmcv-full to mmcv. See https://github.com/open-mmlab/mmcv/blob/master/docs/en/compatibility.md for more details.
warnings.warn(
D:\Anaconda3\Anaconda\envs\op\lib\site-packages\mmdet\models\dense_heads\anchor_head.py:116: UserWarning: DeprecationWarning: `num_anchors` is deprecated, for consistency or also use `num_base_priors` instead
warnings.warn('DeprecationWarning: `num_anchors` is deprecated, '
load checkpoint from local path: oriented_rcnn_r50_fpn_1x_dota_le90-6d2b2ce0.pth
D:\Anaconda3\Anaconda\envs\op\lib\site-packages\mmdet\models\dense_heads\anchor_head.py:123: UserWarning: DeprecationWarning: anchor_generator is deprecated, please use "prior_generator" instead
warnings.warn('DeprecationWarning: anchor_generator is deprecated, '
Traceback (most recent call last):
File "D:\picture\conan\mmrotate\demo\image_demo.py", line 46, in <module>
main(args)
File "D:\picture\conan\mmrotate\demo\image_demo.py", line 33, in main
result = inference_detector(model, args.img)
File "D:\Anaconda3\Anaconda\envs\op\lib\site-packages\mmdet\apis\inference.py", line 157, in inference_detector
results = model(return_loss=False, rescale=True, **data)
File "D:\Anaconda3\Anaconda\envs\op\lib\site-packages\torch\nn\modules\module.py", line 1501, in _call_impl
return forward_call(*args, **kwargs)
File "D:\Anaconda3\Anaconda\envs\op\lib\site-packages\mmcv\runner\fp16_utils.py", line 119, in new_func
return old_func(*args, **kwargs)
File "D:\Anaconda3\Anaconda\envs\op\lib\site-packages\mmdet\models\detectors\base.py", line 174, in forward
return self.forward_test(img, img_metas, **kwargs)
File "D:\Anaconda3\Anaconda\envs\op\lib\site-packages\mmdet\models\detectors\base.py", line 147, in forward_test
return self.simple_test(imgs[0], img_metas[0], **kwargs)
File "d:\picture\conan\mmrotate-0.3.4\mmrotate-0.3.4\mmrotate\models\detectors\two_stage.py", line 179, in simple_test
proposal_list = self.rpn_head.simple_test_rpn(x, img_metas)
File "D:\Anaconda3\Anaconda\envs\op\lib\site-packages\mmdet\models\dense_heads\dense_test_mixins.py", line 130, in simple_test_rpn
proposal_list = self.get_bboxes(*rpn_outs, img_metas=img_metas)
File "D:\Anaconda3\Anaconda\envs\op\lib\site-packages\mmcv\runner\fp16_utils.py", line 208, in new_func
return old_func(*args, **kwargs)
File "d:\picture\conan\mmrotate-0.3.4\mmrotate-0.3.4\mmrotate\models\dense_heads\rotated_rpn_head.py", line 423, in get_bboxes
proposals = self._get_bboxes_single(cls_score_list, bbox_pred_list,
File "d:\picture\conan\mmrotate-0.3.4\mmrotate-0.3.4\mmrotate\models\dense_heads\oriented_rpn_head.py", line 275, in _get_bboxes_single
_, keep = batched_nms(hproposals, scores, ids, cfg.nms)
File "D:\Anaconda3\Anaconda\envs\op\lib\site-packages\mmcv\ops\nms.py", line 350, in batched_nms
dets, keep = nms_op(boxes_for_nms, scores, **nms_cfg_)
File "D:\Anaconda3\Anaconda\envs\op\lib\site-packages\mmcv\utils\misc.py", line 340, in new_func
output = old_func(*args, **kwargs)
File "D:\Anaconda3\Anaconda\envs\op\lib\site-packages\mmcv\ops\nms.py", line 175, in nms
inds = NMSop.apply(boxes, scores, iou_threshold, offset, score_threshold,
File "D:\Anaconda3\Anaconda\envs\op\lib\site-packages\torch\autograd\function.py", line 506, in apply
return super().apply(*args, **kwargs) # type: ignore[misc]
File "D:\Anaconda3\Anaconda\envs\op\lib\site-packages\mmcv\ops\nms.py", line 28, in forward
inds = ext_module.nms(
RuntimeError: nms_impl: implementation for device cuda:0 not found.
这个问题是啥错误啊
```python
代码如下
# Copyright (c) OpenMMLab. All rights reserved.
from argparse import ArgumentParser
from mmdet.apis import inference_detector, init_detector, show_result_pyplot
import mmrotate # noqa: F401
def parse_args():
parser = ArgumentParser()
parser.add_argument('img', help='Image file')
parser.add_argument('config', help='Config file')
parser.add_argument('checkpoint', help='Checkpoint file')
parser.add_argument('--out-file', default=None, help='Path to output file')
parser.add_argument(
'--device', default='cuda:0', help='Device used for inference')
parser.add_argument(
'--palette',
default='dota',
choices=['dota', 'sar', 'hrsc', 'hrsc_classwise', 'random'],
help='Color palette used for visualization')
parser.add_argument(
'--score-thr', type=float, default=0.3, help='bbox score threshold')
args = parser.parse_args()
return args
def main(args):
# build the model from a config file and a checkpoint file
model = init_detector(args.config, args.checkpoint, device=args.device)
# test a single image
result = inference_detector(model, args.img)
# show the results
show_result_pyplot(
model,
args.img,
result,
palette=args.palette,
score_thr=args.score_thr,
out_file=args.out_file)
if __name__ == '__main__':
args = parse_args()
main(args)
```
把mmcv-full卸载再重新安装。
卸载：pip uninstall mmcv-full
安装：pip install mmcv-full
还是不行啊，试过了，可以拜托你帮我看一下是不是版本兼容的问题吗？谢谢~@你好-C嘉嘉
【相关推荐】