Skip to content

yanfeng17/comic_butler

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

12 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

AI 家庭漫画管家

🎨 通过 RTSP 连接家庭监控摄像头,自动抓拍精彩瞬间,AI 打分筛选,生成漫画风格连环画,并推送到微信。

✨ 功能特点

  • 📹 RTSP 抓拍:连接监控摄像头,定时自动抓拍画面
  • 👤 人物检测:AI 智能识别,只保留有人的画面
  • 审美打分:AI 评估图片质量,维护每日 Top N 榜单
  • 🎨 漫画重绘:将照片转换为漫画插画风格
  • 🖼️ 连环画拼接:按时间顺序拼接成漫画分镜
  • 📤 微信推送:通过 PushPlus 推送到微信

🚀 快速开始

1. 安装依赖

cd comic_butler
pip install -r requirements.txt

2. 配置

编辑 config.yaml

# RTSP 摄像头地址
rtsp_url: "rtsp://admin:password@192.168.1.100:554/stream1"

# SiliconFlow API Token (从 https://cloud.siliconflow.cn/i/nSTUhFZV 获取)
siliconflow_token: "your_token_here"

# PushPlus Token (从 http://www.pushplus.plus/ 获取)
pushplus_token: "your_token_here"

# ImgBB API Key (从 https://api.imgbb.com/ 获取,用于图片上传)
imgbb_api_key: "your_key_here"

# 抓拍间隔(秒)
capture_interval: 30

# Top N 榜单数量
top_n: 3

3. 启动

方式一:Streamlit 可视化界面

streamlit run app.py

然后打开浏览器访问 http://localhost:8501

方式二:后台服务模式

python main.py

📋 功能说明

Streamlit 界面

区域 功能
侧边栏 配置 RTSP 地址、API Token、抓拍间隔等参数
控制台 立即抓拍测试、漫画重绘、生成连环画、立即推送
精选展示 今日 Top N 照片预览
日志区 实时查看系统运行状态

API 服务

本项目使用 SiliconFlow 提供的 AI 模型:

  • 人物检测:本地 YOLOv8 (可选)
  • 审美打分THUDM/GLM-4.1V-9B-Thinking(免费模型)
  • 漫画重绘Qwen/Qwen-Image-Edit-2509(0.3元一张图片)

SiliconFlow API 需要通过图片 URL 访问图片,因此必须配置 ImgBB 图床服务:

  • ImgBB:免费图床服务,用于临时托管图片
  • 获取方式:访问 https://api.imgbb.com/ 注册并获取 API Key

推送服务

使用 PushPlus 将漫画连环画推送到微信:

  1. 关注 PushPlus 微信公众号
  2. 获取专属 Token
  3. 填入配置文件

📁 项目结构

comic_butler/
├── app.py                 # Streamlit 界面
├── main.py                # 后台服务入口
├── config_manager.py      # 配置管理
├── vision_client.py       # ModelScope AI 客户端
├── image_utils.py         # 图像处理工具
├── ranking_manager.py     # Top N 排名管理
├── push_client.py         # PushPlus 推送
├── scheduler.py           # 定时任务调度
├── rtsp_capture.py        # RTSP 视频流捕获
├── config.yaml            # 配置文件
├── requirements.txt       # 依赖列表
└── data/                  # 数据目录
    ├── captures/          # 原始抓拍
    ├── cartoons/          # 漫画重绘
    └── collages/          # 拼接长图

⚠️ 注意事项

  1. ImgBB 配置:强烈建议配置 ImgBB API Key,否则 AI 评分和漫画重绘功能将无法正常工作
  2. Token 安全:请勿将 API Token 提交到公开仓库
  3. 网络要求:需要能够访问 SiliconFlow、PushPlus 和 ImgBB API
  4. 摄像头兼容:支持标准 RTSP 协议的摄像头

📝 更新日志

v1.0.0 (2026-01-15)

  • 初始版本发布
  • 支持 RTSP 摄像头连接
  • 集成 SiliconFlow AI 服务
  • 实现漫画连环画生成
  • 添加 PushPlus 微信推送
  • 提供 Streamlit 可视化界面

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors

Languages