input_batch, gt_M_batch, loc_batch = **_gen_batches_**(img, gt, num_samples=FLAGS.n_samples_per_batch, batch_sz=10)