[P] Преобразование изображения в изображение Улучшенный CycleGAN с Resize-Convolution

искусственный интеллект Python GitHub HTTPS

Re-implement CycleGAN in Tensorlayer

Prerequisites:

  • Tensorlayer
  • TensorFlow
  • Python

Run:

CUDA_VISIBLE_DEVICES=0 python main.py 

(if datasets are collected by yourself, you can use dataset_clean.py or dataset_crop.py to pre-process images)

Theory:

The generator process:

Image text

The discriminator process:

Image text

Result Improvement

  • Data augmentation
  • Resize convolution[4]
  • Instance normalization[5]

data augmentation:

Image text

Нормализация экземпляра (сравнение с исходной статьейАР Вест V.org/ABS/1607.08…:

Image text

Resize convolution (Remove Checkerboard Artifacts):

Image text

Image text

Final Results:

Image text

Image text

Reference: