U-NET Light: Validation per class