cfotache / pytorch_objectdetecttrack

Object detection in images, and tracking across video frames
483 stars 253 forks source link

Runtimeerror #24

Open devdastl opened 4 years ago

devdastl commented 4 years ago

whenever I am trying object detection.py I am getting this error- Traceback (most recent call last): File "object_tracker.py", line 87, in tracked_objects = mot_tracker.update(detections.cpu()) File "Z:\DEEP_Learning\original files\Yolo based model\using GPU\pytorch_objectdetecttrack\sort.py", line 208, in update matched, unmatched_dets, unmatched_trks = associate_detections_to_trackers(dets,trks) File "Z:\DEEP_Learning\original files\Yolo based model\using GPU\pytorch_objectdetecttrack\sort.py", line 147, in associate_detections_to_trackers iou_matrix[d,t] = iou(det,trk) RuntimeError: Expected object of type torch.FloatTensor but found type torch.DoubleTensor for argument #3 'other'