if crop_size is None:
      crop_size = [tf.shape(images)[1], tf.shape(images)[2]]