/Cycle-IR-Change

依据版本信息修改过的CycleIR模型

Primary LanguagePython

Cycle-IR: Deep Cyclic Image Retargeting

1. 环境搭建

python环境

python 3.6
numpy 1.19.5
tensorflow 1.11.0
scipy 1.1.0
scikit-learn 0.20.2
scikit-image 0.14.1
opencv-python 3.3.0.10
matplotlib 3.1.2
pillow 8.3.1

2. 使用方法

训练数据集放在 "training data" VGG-16模型放在 "VGG_MODEL"

2.1 使用Circle_reconstruct_image.py来进行测试

2.2 使用CycleIR.py里进行训练，训练好的模型放置在ckpt-wgan文件夹.