ComfyUI-Florence2 项目使用教程
项目地址:https://gitcode.com/gh_mirrors/co/ComfyUI-Florence2
1. 项目的目录结构及介绍
ComfyUI-Florence2/
├── README.md
├── requirements.txt
├── setup.py
├── comfyui/
│ ├── __init__.py
│ ├── main.py
│ ├── config/
│ │ ├── default_config.yaml
│ │ └── custom_config.yaml
│ ├── models/
│ │ ├── florence2.py
│ │ └── __init__.py
│ ├── utils/
│ │ ├── helpers.py
│ │ └── __init__.py
│ └── tests/
│ ├── test_main.py
│ └── __init__.py
└── docs/
├── installation.md
├── usage.md
└── development.md
目录结构说明
README.md
: 项目介绍和基本使用说明。requirements.txt
: 项目依赖文件。setup.py
: 项目安装脚本。comfyui/
: 项目主目录。__init__.py
: 包初始化文件。main.py
: 项目启动文件。config/
: 配置文件目录。default_config.yaml
: 默认配置文件。custom_config.yaml
: 自定义配置文件。
models/
: 模型文件目录。florence2.py
: Florence2 模型实现。
utils/
: 工具函数目录。helpers.py
: 辅助函数文件。
tests/
: 测试文件目录。test_main.py
: 主程序测试文件。
docs/
: 文档目录。installation.md
: 安装指南。usage.md
: 使用指南。development.md
: 开发指南。
2. 项目的启动文件介绍
main.py
main.py
是项目的启动文件,负责初始化配置、加载模型和启动服务。以下是主要功能模块:
import os
from comfyui.config import load_config
from comfyui.models import Florence2
def main():
# 加载配置
config = load_config('comfyui/config/default_config.yaml')
# 初始化模型
model = Florence2(config)
# 启动服务
model.start_service()
if __name__ == "__main__":
main()
功能说明
load_config
: 加载配置文件。Florence2
: 初始化 Florence2 模型。start_service
: 启动模型服务。
3. 项目的配置文件介绍
default_config.yaml
默认配置文件包含了项目运行所需的基本配置参数。以下是部分示例内容:
model:
name: Florence2
version: 2.0
path: comfyui/models/florence2.py
service:
port: 8080
host: 0.0.0.0
logging:
level: INFO
file: logs/comfyui.log
配置参数说明
model
: 模型相关配置。name
: 模型名称。version
: 模型版本。path
: 模型文件路径。
service
: 服务相关配置。port
: 服务端口。host
: 服务地址。
logging
: 日志相关配置。level
: 日志级别。file
: 日志文件路径。
通过以上配置文件,可以灵活调整项目运行时的参数,以适应不同的环境和需求。