face-parsing.PyTorch
face-parsing.PyTorch copied to clipboard
Update transform.py, fix a bug in HorizontalFlip.
fix a bug in HorizontalFlip which deals with the label of right_* and left_*. Please note that the "lb" in "lb == 2" is a PIL object, so if you use "flip_lb[lb == 2] = 3" which changes nothing.