feat: add fire detection training pipeline

This commit is contained in:
2026-08-10 17:04:06 +08:00
parent c71e0be5cc
commit 67d5b40736
11 changed files with 759 additions and 14 deletions

View File

@@ -1,7 +1,7 @@
[project]
name = "yolo"
version = "0.1.0"
description = "Add your description here"
description = "Fire detection training, evaluation, inference, and dataset utilities based on Ultralytics YOLO"
readme = "README.md"
authors = [
{ name = "Kunpeng", email = "mx18782708114@163.com" }
@@ -14,6 +14,7 @@ dependencies = [
"pandas>=3.0.5",
"torch>=2.13.0",
"torchvision>=0.28.0",
"ultralytics>=8.3.0",
]
[tool.uv.sources]
@@ -26,8 +27,8 @@ url = "https://download.pytorch.org/whl/cu132"
explicit = true
[project.scripts]
yolo = "yolo:main"
fire-yolo = "yolo.cli:main"
[build-system]
requires = ["uv_build>=0.12.0,<0.13.0"]
build-backend = "uv_build"
build-backend = "uv_build"