文件名称:pixel-styler:图像到图像翻译的简洁实现
文件大小:12.74MB
文件格式:ZIP
更新时间:2024-05-31 23:46:46
refactoring deep-learning pytorch generative-adversarial-network gan
重构CycleGAN / pix2pix 这是用于图像到图像翻译的的简洁重构版本。 更具可读性和可重用性。 使用具有定制功能闭包的抽象估计器。 在训练过程中使用TensorBoard可视化scalar和image 。 先决条件 Python 3.6+ PyTorch> = 0.3且torchvision > = 0.2 > = 0.3.1 入门 CycleGAN训练/测试 下载CycleGAN数据集(例如地图): bash ./datasets/download_cyclegan_dataset.sh maps 训练模型: python train.py --dataroot ./datasets/maps --name maps_cyclegan --model cycle_gan --no_dropout 要查看训练结果和损失,请运行tensorboard --l
【文件预览】:
pixel-styler-master
----options.py(8KB)
----.flake8(31B)
----train.py(2KB)
----test.py(1KB)
----scripts()
--------test_cyclegan.sh(107B)
--------test_pix2pix.sh(171B)
--------test_single.sh(174B)
--------train_cyclegan.sh(110B)
--------download_pix2pix_dataset.sh(239B)
--------download_cyclegan_dataset.sh(797B)
--------train_pix2pix.sh(212B)
--------combine_A_and_B.py(2KB)
--------download_pix2pix_model.sh(355B)
----main.py(2KB)
----datasets()
--------coco.py(4KB)
--------__init__.py(0B)
--------base_dataset.py(6KB)
--------voc.py(4KB)
--------loader.py(2KB)
----LICENSE(5KB)
----README.md(8KB)
----Makefile(2KB)
----training()
--------networkshd.py(17KB)
--------__init__.py(715B)
--------image_pool.py(1KB)
--------core.py(10KB)
--------trainer.py(15KB)
--------networks.py(15KB)
----docs()
--------real_a_samples.png(704KB)
--------real_b_samples.png(1.64MB)
--------fake_samples_epoch200.png(1.55MB)
--------fake_samples_epoch500.png(1.57MB)
--------horse2zebra.gif(7.33MB)
----.gitignore(226B)