Augment visual inference