feifeiwei / obb-yolov3 Goto Github PK

 /usr/local/lib/python3.6/dist-packages/torch/nn/functional.py:2416: UserWarning: nn.functional.upsample is deprecated. Use nn.functional.interpolate instead.
  warnings.warn("nn.functional.upsample is deprecated. Use nn.functional.interpolate instead.")
---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
<ipython-input-218-382ae974e2fb> in <module>()
      9         optimizer.zero_grad()
     10 
---> 11         losses = model(imgs,targets)  #[loss,x,y,w,h,conf,cls]
     12 
     13         loss = losses[0]

4 frames
/usr/local/lib/python3.6/dist-packages/torch/nn/modules/module.py in __call__(self, *input, **kwargs)
    530             result = self._slow_forward(*input, **kwargs)
    531         else:
--> 532             result = self.forward(*input, **kwargs)
    533         for hook in self._forward_hooks.values():
    534             hook_result = hook(self, input, result)

/content/OBB-YOLOv3/module/model.py in forward(self, x, target)
     96 
     97         if target is None:
---> 98             detections = self.detection(out1, out2, out3)
     99             return detections
    100         else:

/usr/local/lib/python3.6/dist-packages/torch/nn/modules/module.py in __call__(self, *input, **kwargs)
    530             result = self._slow_forward(*input, **kwargs)
    531         else:
--> 532             result = self.forward(*input, **kwargs)
    533         for hook in self._forward_hooks.values():
    534             hook_result = hook(self, input, result)

/content/OBB-YOLOv3/module/layers.py in forward(self, fms, targets)
    103                                                                                                fm_size,
    104                                                                                                self.ignore_threshold,
--> 105                                                                                                self.num_classes)
    106         mask, conf_mask = mask.byte().cuda(), conf_mask.byte().cuda()
    107         tx, ty = tx.cuda(), ty.cuda()

/content/OBB-YOLOv3/module/util.py in get_target(target, anchors, g_dim, ignore_threshold, num_classes)
    137             gw = max(target[b,t,[0,2,4,6]] * g_dim) - min(target[b,t,[0,2,4,6]] * g_dim)
    138             gh = max(target[b,t,[1,3,5,7]] * g_dim) - min(target[b,t,[1,3,5,7]] * g_dim)
--> 139 
    140             # Get shape of gt box
    141             gt_box = torch.FloatTensor(np.array([0, 0, gw, gh])).unsqueeze(0)

TypeError: can't convert np.ndarray of type numpy.object_. The only supported types are: float64, float32, float16, int64, int32, int16, int8, uint8, and bool.