作者： · 发布： · 更新：

电商 AI 生图爆款流水线

核心功能：

本项目实现了一套全自动的 AI 生图流水线,特别适合搞跨境电商的铺货，自动化，和AI辅助提高效率方面：

  1. 以图搜爆款 — 将新品平铺图与历史爆款库做混合检索（视觉相似 + 关键词匹配 + 销量过滤），找到风格最接近的爆款
  2. AI 分析风格 — 将爆款参考图发给大语言模型，自动分析场景、灯光、姿势、氛围，生成风格描述 prompt ，目前基于 nano banana，未来看看加上 gpt-image-2
  3. AI 生成宣传图 — 将新品图 + 爆款参考 + 风格描述一起发给图像生成模型，输出专业宣传图

使用方式很简单：

  1. 将真实商品图片放入 images/（替换占位图），新品放入 new_products/
  2. python main.py setup — 建库 + 编码 + 入库
  3. python main.py search –new-id NEW001 — 测试检索
  4. python main.py generate –new-id NEW001 — 完整生图流水线

技术架构和流程图（借助AI帮忙画出来，佬容易理解）

新品平铺图
    │
    ▼
llama-nemotron-embed-vl-1b-v2 (Embedding API)
    │
    ├── Dense 向量 (2048d)  ──────────────────┐
    │                                          ▼
    └── 文本 → TF-IDF → Sparse 向量 ──→  Milvus 混合检索
                                              │
                                   Dense (视觉相似)
                                +  Sparse (关键词匹配)
                                +  Scalar (品类 + 销量过滤)
                                +  RRF 重排序
                                              │
                                              ▼
                                    Top-K 相似爆款宣传图
                                              │
                                              ▼
                                   Qwen 3.5 (多模态 LLM)
                                   分析场景/灯光/姿势/氛围
                                   输出风格描述 prompt
                                              │
                                              ▼
                                   Gemini 3.1 Flash / Nano Banana 2
                                   (新品图 + 爆款参考 + 风格prompt)
                                              │
                                              ▼
                                    生成的宣传图 → output/

总结：

总体上，就是最主流的AI技术，这些技术是必修的，必用的，所以没有多说，佬友如果有帮助，就开心。

开源地址：

github.com

【开源】电商 AI 生图爆款流水线 – 实现了一套全自动的 AI 生图流水线 – 向量数据库和nano-banana

GitHub – liangdabiao/Fashion-AI: 电商 AI 生图爆款流水线 — 输入新品平铺图，自动检索相似爆款、分析风格，生成专业宣传图。 ##…

电商 AI 生图爆款流水线 — 输入新品平铺图，自动检索相似爆款、分析风格，生成专业宣传图。 ## 项目简介 跨境电商场景中，商家每出一款新品都需要制作宣传图（模特穿着商品的照片）。传统方式需要请模特、搭场景、拍摄修图，成本高、周期长。 本项目实现了一套全自动的 AI 生图流水线

附带一个生成效果图：

开源】电商 AI 生图爆款流水线 – 实现了一套全自动的 AI 生图流水线 – 向量数据库和nano-banana 配图

相关阅读和学习路线

如果你想继续沿着“GitHub 开源项目”这个方向学习，可以先从下面这些站内内容建立路线，再回到本文判断具体资源是否适合自己。

📖 相关推荐

更多资源内容 →
颜资源站长
颜资源站长 已发布 490 篇文章

资深互联网从业者，专注AI工具研究与实战应用。长期跟踪ChatGPT、Claude、Stable Diffusion等前沿AI技术，擅长将复杂的技术概念转化为通俗易懂的教程。运营颜资源小站，致力于为中文用户提供高质量的AI教程、开源项目推荐和数字资源整理。