code problem

[Giveupfree]

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

[tongtybj]

I think you are right... This is a fatal bug, but fortunately, filp augmentation was not used in our training.