微微舟 2021-07-07 20:03 采纳率: 50%
浏览 30

fasterrcnn运行predict.py出了问题是cuda版本出了问题吗

Traceback (most recent call last):
File "g:\maskrcnn\faster-rcnn-pytorch-master\predict.py", line 53, in
r_image = frcnn.detect_image(image)
File "g:\maskrcnn\faster-rcnn-pytorch-master\frcnn.py", line 125, in detect_image
roi_cls_locs, roi_scores, rois, _ = self.model(images)
File "E:\Anaconda3\envs\pytorch\lib\site-packages\torch\nn\modules\module.py", line 547, in call
result = self.forward(*input, **kwargs)
File "g:\maskrcnn\faster-rcnn-pytorch-master\nets\frcnn.py", line 53, in forward
base_feature = self.extractor(x)
File "E:\Anaconda3\envs\pytorch\lib\site-packages\torch\nn\modules\module.py", line 547, in call
result = self.forward(*input, **kwargs)
File "E:\Anaconda3\envs\pytorch\lib\site-packages\torch\nn\modules\container.py", line 92, in forward
input = module(input)
File "E:\Anaconda3\envs\pytorch\lib\site-packages\torch\nn\modules\module.py", line 547, in call
result = self.forward(*input, **kwargs)
File "E:\Anaconda3\envs\pytorch\lib\site-packages\torch\nn\modules\conv.py", line 343, in forward
return self.conv2d_forward(input, self.weight)
File "E:\Anaconda3\envs\pytorch\lib\site-packages\torch\nn\modules\conv.py", line 340, in conv2d_forward
self.padding, self.dilation, self.groups)
Input type (torch.cuda.DoubleTensor) and weight type (torch.cuda.FloatTensor) should be the same

哪里出了问题啊

  • 写回答

1条回答 默认 最新

  • python收藏家 2021-07-08 12:31
    关注

    应该是数据格式问题吧,要统一转换下,都转成doubletensor 试试

    评论

报告相同问题?

悬赏问题

  • ¥20 指导如何跑通以下两个Github代码
  • ¥15 大家知道这个后备文件怎么删吗,为啥这些文件我只看到一份,没有后备呀
  • ¥15 C++为什么这个代码没报错运行不出来啊
  • ¥15 一道ban了很多东西的pyjail题
  • ¥15 关于#r语言#的问题:如何将生成的四幅图排在一起,且对变量的赋值进行更改,让组合的图漂亮、美观@(相关搜索:森林图)
  • ¥15 C++识别堆叠物体异常
  • ¥15 微软硬件驱动认证账号申请
  • ¥15 GPT写作提示指令词
  • ¥20 根据动态演化博弈支付矩阵完成复制动态方程求解和演化相图分析等
  • ¥15 华为超融合部署环境下RedHat虚拟机分区扩容问题