Object localization prediction layer