import tensorflow as tf

    
    mask_ = tf.ones([FLAGS.batch_size,64,64,3])
    mask = tf.pad(mask_, [[0,0],[32,32],[32,32],[0,0]])

    mask2__ = tf.ones([FLAGS.batch_size,78,78,3])
    mask2_ = tf.pad(mask2__, [[0,0],[25,25],[25,25],[0,0]])
    mask2 = mask2_ - mask

    pred_annotation, logits = inference((1-mask)*image + mask*255, keep_probability,z)

