luminoth

GitHub
2.4k 395 较难 1 次阅读 昨天BSD-3-Clause图像开发框架
AI 解读 由 AI 自动生成,仅供参考

Luminoth 是一个基于 Python 构建的开源计算机视觉深度学习工具包,旨在简化物体检测等任务的开发与实验流程。它底层依托强大的 TensorFlow 和 Sonnet 框架,为开发者提供了实现 Faster R-CNN 和 SSD 等经典算法的便捷途径,并支持在 COCO 和 Pascal 等主流数据集上使用预训练模型。

该工具主要解决了研究人员和工程师在复现前沿视觉算法时面临的环境配置复杂、代码复用率低等痛点。通过提供统一的命令行接口(lumi),用户可以轻松完成从数据集适配、本地或云端(如 Google Cloud)模型训练到结果可视化的全流程,极大地降低了技术门槛。其设计特别注重训练过程的透明度,内置了丰富的图表与摘要可视化功能,帮助用户深入理解模型内部的运行状态。

Luminoth 非常适合具备一定编程基础的 AI 开发者、计算机视觉研究人员以及希望快速验证算法原型的团队使用。虽然目前项目已停止维护,官方建议转向 Detectron2 等更新的平台,但其清晰的架构设计和对核心检测算法的完整实现,仍使其成为学习深度学习目标检测原理的优秀参考资源。

使用场景

某电商初创公司的算法团队需要快速构建一个自动识别商品图片中瑕疵(如划痕、污渍)的检测系统,以替代低效的人工质检流程。

没有 luminoth 时

  • 模型复现成本高:工程师需从零编写 Faster R-CNN 或 SSD 等复杂算法的底层代码,耗费数周时间调试网络结构。
  • 数据适配繁琐:缺乏统一的数据集接口,每次更换标注格式(如从 VOC 转 COCO)都需要手动重写数据加载脚本。
  • 训练监控困难:缺少内置的可视化组件,难以直观观察损失函数收敛情况或检测框的实时预测效果。
  • 部署门槛高:没有预训练的权重文件支持,必须收集海量瑕疵图片从头训练,导致项目启动周期漫长。

使用 luminoth 后

  • 开箱即用模型:直接调用 luminoth 内置的 Faster R-CNN 和 SSD 模型,通过简单的命令行配置即可启动训练,将开发周期缩短至几天。
  • 标准化数据流程:利用其灵活的数据集适配工具,轻松导入自定义的瑕疵标注数据,无需关心底层数据解析细节。
  • 直观结果可视化:借助自带的总结图表和图像可视功能,实时监控训练过程中的检测框变化,快速定位模型过拟合或欠拟合问题。
  • 迁移学习加速:直接使用在 COCO 等大型数据集上预训练的检查点(Checkpoints),仅需少量瑕疵样本微调即可达到高精度,大幅降低数据依赖。

luminoth 通过提供标准化的深度学习流程和预训练资源,让计算机视觉团队能从重复造轮子中解放出来,专注于解决具体的业务检测难题。

运行环境要求

GPU
  • 非必需
  • 若需 GPU 加速,需安装 tensorflow-gpu(具体显卡型号、显存大小及 CUDA 版本未在文档中说明,取决于所安装的 TensorFlow 版本要求)
内存

未说明

依赖
notes该项目自 2020 年 1 月起已停止维护,作者建议切换至 Detectron2。目前仅为 Alpha 版本,接口可能随时变更。支持在 Google Cloud ML Engine 上训练(需额外安装 gcloud 依赖)。支持 Faster R-CNN 和 SSD 模型。
python2.7, 3.4, 3.5, 3.6
tensorflow
sonnet
luminoth hero image

快速开始

Luminoth


(2020年1月) Luminoth 已不再维护。我们建议切换到 Facebook 的 Detectron2,它实现了更现代的算法,并支持更多用例。


构建状态 文档状态 codecov 许可证

Luminoth 是一个用于 计算机视觉 的开源工具包。目前我们支持目标检测,但我们的目标是实现更多功能。它基于 Python 构建,使用了 TensorFlowSonnet

完整的文档请参阅 这里

使用 Faster R-CNN 进行目标检测的示例

免责声明:Luminoth 仍处于 Alpha 阶段,这意味着随着代码库的不断完善,内部和外部接口(例如命令行界面)很可能会发生变化。

安装

Luminoth 目前支持 Python 2.7 和 3.4–3.6。

先决条件

要使用 Luminoth,必须先安装 TensorFlow。如果您需要 GPU 支持,请使用 pip install tensorflow-gpu 安装 GPU 版本的 TensorFlow;否则可以使用 pip install tensorflow 安装 CPU 版本。

安装 Luminoth

只需从 PyPI 安装即可:

pip install luminoth

可选地,您也可以通过运行 pip install luminoth[tf]pip install luminoth[tf-gpu] 来让 Luminoth 自动为您安装 TensorFlow,具体取决于您希望使用的 TensorFlow 版本。

Google Cloud

如果您希望使用 Google Cloud ML Engine 进行训练,则需要安装可选依赖项:

pip install luminoth[gcloud]

从源码安装

首先在您的机器上克隆仓库,然后使用 pip 安装:

git clone https://github.com/tryolabs/luminoth.git
cd luminoth
pip install -e .

检查安装是否成功

只需运行 lumi --help 即可。

支持的模型

目前我们支持以下模型:

我们计划在不久的将来添加对更多模型的支持,例如 RetinaNetMask R-CNN

我们还提供了上述模型的 预训练检查点,这些检查点是在流行的 数据集 上训练得到的,例如 COCOPascal

使用方法

有一个主要的命令行界面,可以通过 lumi 命令使用。当您不确定如何操作时,只需输入:

lumi --helplumi <子命令> --help

就会显示可用选项及其描述列表。

处理数据集

请参阅 适配数据集

训练

请参阅 训练您自己的模型,了解如何在本地或 Google Cloud 中进行训练。

可视化结果

我们致力于生成有用且易于理解的摘要和图表可视化。我们认为这些不仅对于监控至关重要(当然如此!),而且有助于更全面地理解系统内部的工作原理。就像代码需要清晰易懂一样,计算图也应该同样直观易读。

默认情况下,摘要和图日志会保存在当前目录下的 jobs/ 文件夹中。您可以使用 TensorBoard 查看这些日志,运行以下命令:

tensorboard --logdir path/to/jobs

为什么叫这个名字?

黑暗面罩是《银河战士Prime 2:回音》中的面罩升级版。它由 卢米诺斯族 在战争期间设计,曾被以太界的冠军 A-Kul 用来穿透黑暗以太界的迷雾,与英格族作战。

—— 黑暗面罩 - Wikitroid

许可证

版权所有 © 2018, Tryolabs。 根据 BSD 3-Clause 许可证发布。

版本历史

v0.2.32018/11/09
v0.2.22018/11/07
v0.2.12018/10/31
v0.2.02018/08/24
v0.1.22018/06/26
v0.1.12018/04/17
v0.1.02018/03/21
v0.0.32018/02/14
v0.0.22017/11/17
v0.0.12017/10/10

常见问题

相似工具推荐

openclaw

OpenClaw 是一款专为个人打造的本地化 AI 助手,旨在让你在自己的设备上拥有完全可控的智能伙伴。它打破了传统 AI 助手局限于特定网页或应用的束缚,能够直接接入你日常使用的各类通讯渠道,包括微信、WhatsApp、Telegram、Discord、iMessage 等数十种平台。无论你在哪个聊天软件中发送消息,OpenClaw 都能即时响应,甚至支持在 macOS、iOS 和 Android 设备上进行语音交互,并提供实时的画布渲染功能供你操控。 这款工具主要解决了用户对数据隐私、响应速度以及“始终在线”体验的需求。通过将 AI 部署在本地,用户无需依赖云端服务即可享受快速、私密的智能辅助,真正实现了“你的数据,你做主”。其独特的技术亮点在于强大的网关架构,将控制平面与核心助手分离,确保跨平台通信的流畅性与扩展性。 OpenClaw 非常适合希望构建个性化工作流的技术爱好者、开发者,以及注重隐私保护且不愿被单一生态绑定的普通用户。只要具备基础的终端操作能力(支持 macOS、Linux 及 Windows WSL2),即可通过简单的命令行引导完成部署。如果你渴望拥有一个懂你

349.3k|★★★☆☆|1周前
Agent开发框架图像

stable-diffusion-webui

stable-diffusion-webui 是一个基于 Gradio 构建的网页版操作界面,旨在让用户能够轻松地在本地运行和使用强大的 Stable Diffusion 图像生成模型。它解决了原始模型依赖命令行、操作门槛高且功能分散的痛点,将复杂的 AI 绘图流程整合进一个直观易用的图形化平台。 无论是希望快速上手的普通创作者、需要精细控制画面细节的设计师,还是想要深入探索模型潜力的开发者与研究人员,都能从中获益。其核心亮点在于极高的功能丰富度:不仅支持文生图、图生图、局部重绘(Inpainting)和外绘(Outpainting)等基础模式,还独创了注意力机制调整、提示词矩阵、负向提示词以及“高清修复”等高级功能。此外,它内置了 GFPGAN 和 CodeFormer 等人脸修复工具,支持多种神经网络放大算法,并允许用户通过插件系统无限扩展能力。即使是显存有限的设备,stable-diffusion-webui 也提供了相应的优化选项,让高质量的 AI 艺术创作变得触手可及。

162.1k|★★★☆☆|1周前
开发框架图像Agent

everything-claude-code

everything-claude-code 是一套专为 AI 编程助手(如 Claude Code、Codex、Cursor 等)打造的高性能优化系统。它不仅仅是一组配置文件,而是一个经过长期实战打磨的完整框架,旨在解决 AI 代理在实际开发中面临的效率低下、记忆丢失、安全隐患及缺乏持续学习能力等核心痛点。 通过引入技能模块化、直觉增强、记忆持久化机制以及内置的安全扫描功能,everything-claude-code 能显著提升 AI 在复杂任务中的表现,帮助开发者构建更稳定、更智能的生产级 AI 代理。其独特的“研究优先”开发理念和针对 Token 消耗的优化策略,使得模型响应更快、成本更低,同时有效防御潜在的攻击向量。 这套工具特别适合软件开发者、AI 研究人员以及希望深度定制 AI 工作流的技术团队使用。无论您是在构建大型代码库,还是需要 AI 协助进行安全审计与自动化测试,everything-claude-code 都能提供强大的底层支持。作为一个曾荣获 Anthropic 黑客大奖的开源项目,它融合了多语言支持与丰富的实战钩子(hooks),让 AI 真正成长为懂上

158.1k|★★☆☆☆|今天
开发框架Agent语言模型

ComfyUI

ComfyUI 是一款功能强大且高度模块化的视觉 AI 引擎,专为设计和执行复杂的 Stable Diffusion 图像生成流程而打造。它摒弃了传统的代码编写模式,采用直观的节点式流程图界面,让用户通过连接不同的功能模块即可构建个性化的生成管线。 这一设计巧妙解决了高级 AI 绘图工作流配置复杂、灵活性不足的痛点。用户无需具备编程背景,也能自由组合模型、调整参数并实时预览效果,轻松实现从基础文生图到多步骤高清修复等各类复杂任务。ComfyUI 拥有极佳的兼容性,不仅支持 Windows、macOS 和 Linux 全平台,还广泛适配 NVIDIA、AMD、Intel 及苹果 Silicon 等多种硬件架构,并率先支持 SDXL、Flux、SD3 等前沿模型。 无论是希望深入探索算法潜力的研究人员和开发者,还是追求极致创作自由度的设计师与资深 AI 绘画爱好者,ComfyUI 都能提供强大的支持。其独特的模块化架构允许社区不断扩展新功能,使其成为当前最灵活、生态最丰富的开源扩散模型工具之一,帮助用户将创意高效转化为现实。

108.3k|★★☆☆☆|6天前
开发框架图像Agent

gemini-cli

gemini-cli 是一款由谷歌推出的开源 AI 命令行工具,它将强大的 Gemini 大模型能力直接集成到用户的终端环境中。对于习惯在命令行工作的开发者而言,它提供了一条从输入提示词到获取模型响应的最短路径,无需切换窗口即可享受智能辅助。 这款工具主要解决了开发过程中频繁上下文切换的痛点,让用户能在熟悉的终端界面内直接完成代码理解、生成、调试以及自动化运维任务。无论是查询大型代码库、根据草图生成应用,还是执行复杂的 Git 操作,gemini-cli 都能通过自然语言指令高效处理。 它特别适合广大软件工程师、DevOps 人员及技术研究人员使用。其核心亮点包括支持高达 100 万 token 的超长上下文窗口,具备出色的逻辑推理能力;内置 Google 搜索、文件操作及 Shell 命令执行等实用工具;更独特的是,它支持 MCP(模型上下文协议),允许用户灵活扩展自定义集成,连接如图像生成等外部能力。此外,个人谷歌账号即可享受免费的额度支持,且项目基于 Apache 2.0 协议完全开源,是提升终端工作效率的理想助手。

100.8k|★★☆☆☆|1周前
插件Agent图像

markitdown

MarkItDown 是一款由微软 AutoGen 团队打造的轻量级 Python 工具,专为将各类文件高效转换为 Markdown 格式而设计。它支持 PDF、Word、Excel、PPT、图片(含 OCR)、音频(含语音转录)、HTML 乃至 YouTube 链接等多种格式的解析,能够精准提取文档中的标题、列表、表格和链接等关键结构信息。 在人工智能应用日益普及的今天,大语言模型(LLM)虽擅长处理文本,却难以直接读取复杂的二进制办公文档。MarkItDown 恰好解决了这一痛点,它将非结构化或半结构化的文件转化为模型“原生理解”且 Token 效率极高的 Markdown 格式,成为连接本地文件与 AI 分析 pipeline 的理想桥梁。此外,它还提供了 MCP(模型上下文协议)服务器,可无缝集成到 Claude Desktop 等 LLM 应用中。 这款工具特别适合开发者、数据科学家及 AI 研究人员使用,尤其是那些需要构建文档检索增强生成(RAG)系统、进行批量文本分析或希望让 AI 助手直接“阅读”本地文件的用户。虽然生成的内容也具备一定可读性,但其核心优势在于为机器

93.4k|★★☆☆☆|1周前
插件开发框架