feat: add fire detection training pipeline
This commit is contained in:
@@ -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"
|
||||
Reference in New Issue
Block a user