diff --git a/datasets/linemod/dataset.py b/datasets/linemod/dataset.py index b887a81c60ef019a3eed778ebbaf5120f117f845..0a3a852ca5417c4790f68dd50383ca2c861dab97 100755 --- a/datasets/linemod/dataset.py +++ b/datasets/linemod/dataset.py @@ -215,7 +215,9 @@ img_length = 640 def mask_to_bbox(mask): mask = mask.astype(np.uint8) - _, contours, _ = cv2.findContours(mask, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE) + contours, _ = cv2.findContours(mask, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE) + + x = 0 y = 0 w = 0