Merge pull request #1 from hzzh11856/type-convertor

fix bug of "Expected object of scalar type Float but got scalar type Double for argument open-mmlab#2 'other'"
hzzh11856 · Oct 27, 2019 · 11fc275 · 11fc275
2 parents 82c533b + b1f4c34
commit 11fc275
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/mmdet/core/bbox/geometry.py b/mmdet/core/bbox/geometry.py
@@ -21,6 +21,7 @@ def bbox_overlaps(bboxes1, bboxes2, mode='iou', is_aligned=False):
 
     assert mode in ['iou', 'iof']
 
+    bboxes2 = bboxes2.type(bboxes1.dtype)
     rows = bboxes1.size(0)
     cols = bboxes2.size(0)
     if is_aligned: