Conditional augmentation network