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

本帖使用社区开源推广,符合推广要求。我申明并遵循社区要求的以下内容: 我的帖子已经打上 开源推广 标签: 是 我的开源项目完整开源,无未开源部分: 是 我的开源项目已链接认可 LINUX DO 社区: 是 我帖子内的项目介绍,AI生成、润色内容部分已截图发出: 是 以上选择我承诺是永久有效的,接受社...
【开源】电商 AI 生图爆款流水线 - 实现了一套全自动的 AI 生图流水线 - 向量数据库和nano-banana
【开源】电商 AI 生图爆款流水线 - 实现了一套全自动的 AI 生图流水线 - 向量数据库和nano-banana
本帖使用社区开源推广,符合推广要求。我申明并遵循社区要求的以下内容:
  • 我的帖子已经打上 开源推广 标签:
  • 我的开源项目完整开源,无未开源部分:
  • 我的开源项目已链接认可 LINUX DO 社区:
  • 我帖子内的项目介绍,AI生成、润色内容部分已截图发出:
  • 以上选择我承诺是永久有效的,接受社区和佬友监督:

以下为项目介绍正文内容,AI生成、润色内容已使用截图方式发出


电商 AI 生图爆款流水线

开源一个实用价值的电商项目给佬友,因为linux.do里面很多人搞跨境电商的,希望多多支持,希望有帮助,我自己觉得非常不错的,所以推荐给佬友:

核心功能:

本项目实现了一套全自动的 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

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

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

附带一个生成效果图:

image

3 个帖子 - 3 位参与者

阅读完整话题

来源: linux.do查看原文