HiDT 快速入门:10 分钟用命令行把白天照片变成日落与夜景
HiDT 快速入门10 分钟用命令行把白天照片变成日落与夜景【免费下载链接】HiDTOfficial repository for the paper High-Resolution Daytime Translation Without Domain Labels (CVPR2020, Oral)项目地址: https://gitcode.com/gh_mirrors/hi/HiDT想不想把一张普普通通的白天照片一键变成橙红夕阳、深邃夜景甚至璀璨星空HiDTHigh-Resolution Daytime Translation正是为此而生的开源 AI 图像转换工具——它在不需要任何领域标签的前提下就能把白天照片转成日落与夜景等风格并输出高分辨率成品。本文是一份 HiDT 快速入门教程手把手带你 10 分钟完成环境安装与命令行推理把白天照片转夜景变成一行命令的事。HiDT 是什么为什么它能做出自然夜景HiDT 是 CVPR 2020 Oral 论文《High-Resolution Daytime Translation Without Domain Labels》的官方 PyTorch 实现。它的核心思路是把图像解耦为“内容”与“风格”两部分内容编码器提取场景结构风格编码器提取光线与色调解码器通过 AdaIN 把日落、夜晚的风格注入白天照片再配合一个轻量增强网络G_enh把 256px 结果升级到 1024×1024 高清画质。正因为内容与风格分离HiDT 才能做到“一张白天图 任意风格参考图 自然逼真的新照片”而不是生硬的调色滤镜。推理核心类见 hidt/style_transformer.py增强网络见 hidt/networks/enhancement/RRDBNet_arch.py。第一步环境安装约 3 分钟HiDT 需要 Python 3.7 及以上推荐使用 conda 创建独立环境避免依赖冲突。复制下面这 4 条命令即可完成“克隆 安装”conda create -n hidt python3.7 conda activate hidt git clone https://gitcode.com/gh_mirrors/hi/HiDT cd HiDT pip install -r requirements.txt依赖非常精简完整清单见 requirements.txt核心只有 PyTorch、torchvision、numpy、PyYAML、Pillow 与 tqdm。克隆下来的仓库已自带示例图片和全部预训练权重无需再下载任何模型。第二步一行命令把白天照片转成日落夜景仓库自带处理脚本infer_on_folders.py只需指定“内容图文件夹”“风格图文件夹”和配置文件即可批量转换。直接复制下面的命令运行CUDA_VISIBLE_DEVICES0 PYTHONPATH$(pwd):${PYTHONPATH} \ python ./bin/infer_on_folders.py \ --content-dir ./images/daytime/content/ \ --style-dir ./images/daytime/styles/ \ --cfg-path ./configs/daytime.yaml \ --chk-path ./trained_models/generator/daytime.pt \ --enh-path ./trained_models/enhancer/enhancer.pth \ --enhancement generator几个关键参数的含义如下参数作用--content-dir白天照片所在文件夹示例见 images/daytime/content/--style-dir日落/夜景风格参考图文件夹示例见 images/daytime/styles/--cfg-path模型配置文件 configs/daytime.yaml--chk-path生成器权重 trained_models/generator/daytime.pt--enh-path高清增强权重 trained_models/enhancer/enhancer.pth--enhancementgenerator启用高清增强fullconv为纯卷积快速模式运行结束后输出目录里就是一张张从白天变身日落与夜景的高清照片。下面的对比图可以直观感受不同场景的转换效果第三步风格插值玩出日落→夜景的丝滑过渡HiDT 最炫的功能之一就是风格插值把两个风格向量比如“日落”与“夜晚”按不同比例混合就能生成从黄昏一路渐变到深夜的连续画面非常适合做延时摄影、短视频素材。同样地它还可以在“阴天”与“晴天”之间做光线插值一键把灰蒙蒙的照片变得通透明亮想手动控制插值比例可以用官方 Colab 版 Notebook 直接改参数试玩入口见 notebooks/HighResolutionDaytimeTranslation.ipynb无需配置本地环境。第四步换种玩法——把照片变成世界名画风格除了日落夜景HiDT 还附带了一个用 WikiArt 画作训练的模型能让你的照片拥有油画、印象派等艺术风格。只需把命令中的配置与权重换成 wikiart 版本CUDA_VISIBLE_DEVICES0 PYTHONPATH$(pwd):${PYTHONPATH} \ python ./bin/infer_on_folders.py \ --content-dir ./images/wikiart/content/ \ --style-dir ./images/wikiart/styles/ \ --cfg-path ./configs/wikiart.yaml \ --chk-path ./trained_models/generator/wikiart.pt \ --enh-path ./trained_models/enhancer/enhancer.pth \ --enhancement generator对应的配置见 configs/wikiart.yaml风格参考图见 images/wikiart/styles/。对比两个配置文件可以发现wikiart 模型的风格编码维度更高output_dim: 12能承载更丰富的艺术风格信息。进阶高分辨率增强管线是怎么工作的为什么 HiDT 能输出高清夜景而不糊秘密在于它的分块增强管线先把低分辨率结果切成若干重叠小块逐块送入增强网络G_enh做超分再由合并网络无缝拼回整图。示意图如下这也解释了为什么 trained_models/enhancer/enhancer.pth 是独立的一个权重文件——它是 HiDT 高清输出的关键一环。常见问题速查Q没有 GPU 能跑吗可以但推理会很慢且 1024×1024 高清增强建议至少 6GB 显存。Q自己的照片怎么用把照片放进--content-dir指向的文件夹即可风格图同理格式支持常见的 jpg/png。Q输出效果有伪影怎么办生成器训练时按 256px 推理建议保持inference_size256更高分辨率反而可能出现伪影。总结从安装到出图HiDT 的全流程不到 10 分钟克隆仓库 → 装依赖 → 跑一条命令。无论是白天照片转日落夜景、风格插值还是艺术风格迁移它都能给出高分辨率、观感自然的结果。赶紧动手试试用命令行把你的白天照片变成夜幕下的美景吧【免费下载链接】HiDTOfficial repository for the paper High-Resolution Daytime Translation Without Domain Labels (CVPR2020, Oral)项目地址: https://gitcode.com/gh_mirrors/hi/HiDT创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考