import os
from PIL import Image
UNIT_SIZE = 220 # the size of image
save_path = '/root/group-dia/zxb/Code-/lip-CycleGAN-and-pix2pix-master/checkpoints/lip_cyclegan_6.0/web/result/out'
path = "/root/group-dia/zxb/Code-/lip-CycleGAN-and-pix2pix-master/checkpoints/lip_cyclegan_6.0/web/images"
images = []
def pinjie(images):
  for i in range(len(images) / 6):
    target = Image.new('RGB', (UNIT_SIZE*3, UNIT_SIZE*2))  # result is 2*3
    leftone = 0
    lefttwo = 0
    rightone = UNIT_SIZE
    righttwo = UNIT_SIZE
    for j in range(6):
      if(j <= 2):
        target.paste(images[j + i*6], (leftone, 0, rightone, UNIT_SIZE))
        leftone += UNIT_SIZE
        rightone += UNIT_SIZE
      else:
        target.paste(images[j + i*6], (lefttwo, UNIT_SIZE, righttwo, UNIT_SIZE*2))
        lefttwo += UNIT_SIZE
        righttwo += UNIT_SIZE
    quality_value = 500
    target.save(save_path + '{}.png'.format(i), quality=quality_value)
if __name__ == '__main__':
  for img in os.listdir(path):
    images.append(Image.open(os.path.join(path, img)))
  print len(images)
  pinjie(images)