code: def main(_):
    
    batch_shape = [FLAGS.batch_size, FLAGS.image_height, FLAGS.image_width, 3]
    for filenames, images in load_images(FLAGS.input_dir, batch_shape):
      save_images(images, filenames, FLAGS.output_dir)
docs: Run the sample attack