winterest
commented
6 years ago Have you figured it out?
Open GYM13 opened 6 years ago
winterest
commented
6 years ago Have you figured it out?
Millielele
commented
3 years ago cuda8.0 pytorch0.4.0 maybe work
Traceback (most recent call last): File "check_model.py", line 101, in
scores, boxes, net_conv, im_scale = im_detect(net, im)
File "../tools/../lib/model/test.py", line 104, in imdetect
, scores, bbox_pred, rois, net_conv = net.test_image(blobs['data'], blobs['im_info'])
File "../tools/../lib/nets/network.py", line 571, in test_image
self.forward(image, im_info, None, None, mode='TEST')
File "../tools/../lib/nets/network.py", line 541, in forward
net_conv, rois, cls_prob, bbox_pred, mask_prob = self._predict()
File "../tools/../lib/nets/network.py", line 468, in _predict
rois = self._region_proposal(net_conv)
File "../tools/../lib/nets/network.py", line 256, in _regionproposal
rois, = self._proposal_layer(rpn_cls_prob, rpn_bbox_pred)
File "../tools/../lib/nets/network.py", line 94, in _proposal_layer
self._feat_stride, self._anchors, self._num_anchors)
File "../tools/../lib/layer_utils/proposal_layer.py", line 61, in proposal_layer
proposals = proposals[keep, :]
File "/usr/lib64/python2.7/site-packages/torch/autograd/variable.py", line 78, in getitem
return Index.apply(self, key)
File "/usr/lib64/python2.7/site-packages/torch/autograd/_functions/tensor.py", line 87, in forward
result = i.index(ctx.index)