Как сохранить одинаковые имена изображений при сохранении их в коде Python?

Question

В этом коде у меня есть тысячи изображений видеопоследовательности, и я хочу, чтобы мой код сохранил их с тем же именем, что и при загрузке его для тестирования.Например, следующий код выполняет тестирование изображений и сохраняет его в последней строке «save_images (.....)» с новым именем «test_0001.png (до 4 десятичных разрядов)», но что мне нужно, так это любой примерimage мой код, выбранный из папки тестирования, например, имя bin2225.png, после тестирования следует сохранить его с тем же точным именем, bin2225.png, что и вывод.Как изменить это в последней строке кода?save_images (........)

def test(self, args):      
    init_op = tf.global_variables_initializer()
    self.sess.run(init_op)

    sample_files = glob('./datasets/{}/val/*.png'.format(self.dataset_name))

    # sort testing input
    n = [i for i in map(lambda x: x.split('/')[-1].split('.png')[0], sample_files)]
    sample_files = [x for (y, x) in sorted(zip(n, sample_files))]

    # load testing input
    print("Loading testing images ...")
    sample = [load_data(sample_file, is_test=True) for sample_file in sample_files]

    if (self.is_grayscale):
        sample_images = np.array(sample).astype(np.float32)[:, :, :, None]
    else:
        sample_images = np.array(sample).astype(np.float32)

    sample_images = [sample_images[i:i+self.batch_size]
                     for i in xrange(0, len(sample_images), self.batch_size)]
    sample_images = np.array(sample_images)
    print(sample_images.shape)

    start_time = time.time()
    if self.load(self.checkpoint_dir):
        print(" [*] Load SUCCESS")
    else:
        print(" [!] Load failed...")

    for i, sample_image in enumerate(sample_images):
        idx = i+1
        print("sampling image ", idx)
        samples = self.sess.run(
            self.fake_B_sample,
            feed_dict={self.real_data: sample_image}
        )
        save_images(samples, [self.batch_size, 1],
                    './{}/test_{:04d}.png'.format(args.test_dir, idx))