code problem
Giveupfree opened this issue · comments
https://github.com/tongtybj/TrTr/blob/master/datasets/augmentation.py
Is line 174 wrong? what is returned on the code is bbox. Shouldn't it be mask?
def _flip_aug(self, image, bbox, mask):
image = cv2.flip(image, 1)
width = image.shape[1]
bbox = Corner(width - 1 - bbox.x2, bbox.y1,width - 1 - bbox.x1, bbox.y2)
bbox = Corner(width - 1 - mask[2], mask[1],width - 1 - mask[0], mask[3])
return image, bbox, mask
I think you are right...
This is a fatal bug, but fortunately, filp
augmentation was not used in our training.