vishalrishu
commented
3 years ago I'm getting same error. Please let me know if you get the solution to this query. Thanks @buaaswf
Open buaaswf opened 3 years ago
vishalrishu
commented
3 years ago I'm getting same error. Please let me know if you get the solution to this query. Thanks @buaaswf
FloatyDev
commented
2 months ago has anyone found the solution to this?
File "train_FN.py", line 400, in
main()
File "train_FN.py", line 279, in main
use_gt_boxes=args.use_gt_boxes)
File "/home/swf/F-Net/models/HDN_v2/engines_v1.py", line 133, in test
use_gt_boxes=use_gt_boxes)
File "/home/swf/F-Net/models/HDN_v2/factorizable_network_v4.py", line 280, in evaluate
object_result, predicate_result = self.forward_eval(im_data, im_info,)
File "/home/swf/F-Net/models/HDN_v2/factorizable_network_v4.py", line 232, in forward_eval
features, objectrois, = self.rpn(im_data, im_info)
File "/home/swf/anaconda3/envs/f-net/lib/python2.7/site-packages/torch/nn/modules/module.py", line 489, in call
result = self.forward(*input, **kwargs)
File "/home/swf/F-Net/models/RPN/RPN.py", line 115, in forward
mappings=self.opts['mappings'])
File "/home/swf/F-Net/models/RPN/RPN.py", line 141, in proposal_layer
_feat_stride, opts, anchor_scales, anchor_ratios, mappings)
File "/home/swf/F-Net/lib/rpn_msr/proposal_layer.py", line 134, in proposal_layer
keep = nms(np.hstack((proposals, scores)).astype(np.float32), nms_thres)
File "/home/swf/F-Net/lib/fast_rcnn/nms_wrapper.py", line 26, in nms
return nms_gpu(dets[:, :4], dets[:, 4], thresh).cpu().numpy()
RuntimeError: Not compiled with GPU support (nms at /home/dell/F-Net/lib/layer_utils/csrc/nms.h:22)
frame #0: c10::Error::Error(c10::SourceLocation, std::string const&) + 0x45 (0x7fc66e615cf5 in /home/swf/anaconda3/envs/f-net/lib/python2.7/site-packages/torch/lib/libc10.so)