AnimeGAN

「Open Source」. The pytorch implementation of AnimeGAN.
「动画风格生成,用AI拍出一部属于自己的动漫」

数据集

下载地址
训练数据结构设计

|_datasets  
|__animegan  
|___Shinkai(画家风格)  
|_____train(训练数据)  
|________real(真实数据)  
|________style(风格数据)  
|________smooth(经过处理的风格数据)  
|_____test(测试数据)  
|________real(真实数据)  

训练

# 单卡
python tools/train_net.py \
--config-file "path/to/config" \
SOLVER.IMS_PER_BATCH 8

# 多卡
python -m torch.distributed.launch --nproc_per_node=8 \
/tools/train_net.py \
--config-file "path/to/config" \
SOLVER.IMS_PER_BATCH 8

模型下载

图片转换

# --config-path 配置文件路径
# --image 图像路径
# MODEL.WEIGHT 模型文件路径
python scripts/image2anime.py \
--config-file "your_config_path" \
--image "your_image_path" \
MODEL.WEIGHT "your_model_path"

视频转换

# --config-path 配置文件路径
# --video 视频路径
# MODEL.WEIGHT 模型文件路径
python scripts/image2anime.py \
--config-file "your_config_path" \
--video "your_video_path" \
MODEL.WEIGHT "your_model_path"

demo

😍 Photo to Shinkai Style 😍 Photo to Hayao Style

兴趣小组

添加好友,申请进群

Thanks

Thanks for TachibanaYoshino