code problem

[Huiyiqianli]

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

[趙 漠居(Zhao, Moju)]

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