PITI

用预训练模型,小样本也能搞定高质量图像翻译

开源 🏷️完全免费 🔧 仅开源项目,无在线服务,需自行部署 图像生成 🍴 复刻 26 👀 关注 24
GitHub ↗
GitHub 星标 ★ 504
维护状态 低维护
是否开源 开源
定价模式 🏷️完全免费 开源项目,MIT许可证,完全免费,可自行部署使用。

项目预览

PITI 项目界面预览图
图片来源:GitHub 官方仓库名片

项目数据

分类图像生成
开发团队PITI-Synthesis
定价模式🏷️完全免费
价格说明开源项目,MIT许可证,完全免费,可自行部署使用。
访问状态🔧 仅开源项目,无在线服务,需自行部署
是否开源是
开源协议MIT
主要语言Python
技术栈/模型computer-vision,image-generation,image-synthesis,image-to-image-translation,pretraining
GitHub 星标★ 504
上线时间2022-05-27 00:00:00
代码更新2024-06-02
仓库体积1.4 MB
本站数据更新2026-11-10
维护状态低维护
收录时间2026-08-09
浏览次数4

项目介绍

PITI是一个基于预训练的图像到图像翻译框架,核心思想是“预训练即一切”,旨在解决传统图像翻译模型需要大量配对数据、训练成本高且泛化能力弱的问题。它利用大规模预训练模型(如CLIP)作为特征提取器,通过轻量级适配层将预训练知识迁移到下游翻译任务,从而在少量样本下实现高质量的图像转换。项目支持多种翻译任务,如语义分割图转真实图像、边缘图转照片等,并提供了预训练模型和推理代码。其核心能力包括:高效的迁移学习、强大的语义理解、以及跨域一致性保持。PITI在多个基准数据集上展示了优于现有方法的性能,尤其在小样本场景下表现突出,为图像合成领域提供了一种新的范式。

Star 增长趋势

本站每日记录该项目 GitHub 星标变化,观测越久曲线越完整

数据来源:GitHub API 逐日快照 · 收录于 2026-08-09 · 当前星标 ★ 504

使用教程

难度:高级 约 60 分钟 部署方式:本地GPU 5 步

环境要求

  • CUDA 显卡
  • Python 3.10+
  • 较大显存

安装与启动步骤

  1. 1拉仓库装依赖

    按清单装。

    git clone https://github.com/PITI-Synthesis/PITI.git
  2. 2装界面依赖

    它带一个网页交互界面。

    pip install gradio
  3. 3跑推理

    先跑单样本看效果。

    python inference.py
  4. 4跑批量采样

    批量脚本已给。

    bash sample.sh
  5. 5要微调再跑训练脚本

    微调分两阶段,顺序不能反。

    bash mask_finetune_base.sh

如何确认成功

用一张参考图加一个布局条件生成结果,输出在姿态与外观上都符合给定条件

注意事项

  • 它做的是带布局条件的图像生成,把结构控制与外观生成分开。
  • 微调分基础与上采样两阶段,顺序反了效果会崩。
  • 推理与微调脚本都给了,先跑推理确认环境。
  • 显存需求高,批量采样注意并发数。

核心亮点

  • 基于CLIP预训练,语义理解强,翻译结果更逼真
  • 小样本即可微调,大幅降低数据标注成本
  • 支持多任务通用框架,代码结构清晰易扩展

不足之处

  • 依赖大规模预训练模型,推理资源需求较高
  • 文档/社区待观察

适用场景

  • 游戏美术:根据草图或语义图快速生成概念图
  • 电商设计:将产品线稿自动转换为写实效果图
  • 科研实验:在少量标注数据下进行图像风格迁移

替代项目

CycleGAN、pix2pix、ControlNet

上一篇:PIDM

下一篇:stable-diffusion-prompt-reader

同类项目推荐

gowall 开源

一条命令提取壁纸配色,顺带压缩、OCR 和放大图片

A tool to convert a Wallpaper's color scheme / palette, OCR with VLM's Traditional &···

★ 2332 2026-11-10
ai-picture-editor 开源

说句话就能修图,一周学完 AI Agent 全栈

2026 年编程导航 AI 编程实战新项目,基于 React 19 + FastAPI + LangChain + LangGraph +···

★ 307 2026-09-25
blur-diffusion 开源

渐进去模糊,让扩散模型生成更快更清晰

Official PyTorch implementation of the paper Progressive Deblurring of Diffusion Mod···

★ 159 2026-08-09
FreeScale 开源

免训练提升扩散模型分辨率,一张图生成高清细节

[ICCV 2025] Code for FreeScale, a tuning-free method for higher-resolution visual ge···

★ 148 2026-08-09
ai-draw 开源

说句话就能画图改图,草图流程图随手出

Create and edit drawings through conversation.

★ 974 2026-10-31
Anything2Image 开源

把文字、声音、图片统统变成图像,多模态生成一步到位。

Generate image from anything with ImageBind and Stable Diffusion

★ 202 2026-08-09
qiaomu-codex-imagegen 开源

一个接口让任意 Agent 调用 Codex 生图,附小红书海报模板

让任何 Agent 调用 Codex 内置生图(MCP + CLI + Skill),内置小红书、视频封面、Mondo 海···

★ 136 2026-10-23
diffusionmagic 开源

用简单代码跑通 Stable Diffusion 工作流,快速出图

Easy to use Stable diffusion workflows using diffusers

★ 122 2026-08-09
FontDiffuser 开源

一个参考字,AI帮你生成整套同风格字体

[AAAI2024] FontDiffuser: One-Shot Font Generation via Denoising Diffusion with Multi···

★ 579 2026-08-09