-
Notifications
You must be signed in to change notification settings - Fork 4
/
pyproject.toml
38 lines (34 loc) · 1 KB
/
pyproject.toml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
[tool.poetry]
name = "YOEO"
version = "1.6.0"
description = "A hybrid CNN for object detection and semantic segmentation"
authors = ["Florian Vahl <[email protected]>", "Jan Gutsche <[email protected]>"]
[tool.poetry.dependencies]
python = ">=3.10,<4.0"
matplotlib = "^3.3.3"
terminaltables = "^3.1.10"
imgaug = "^0.4.0"
numpy = "^1.26"
torch = "^2.5.1"
torchvision = "^0.20.1"
pillow = "^11.0.0"
torchsummary = "^1.5.1"
tqdm = "^4.67.0"
opencv-python = "^4.10.0.84"
pyyaml = "^6.0.2"
tensorboard = "^2.18.0"
[tool.poetry.group.dev.dependencies]
onnxruntime = "^1.14.0"
profilehooks = "^1.12.0"
onnx = "^1.9.0"
onnx-simplifier = "^0.3.5"
[build-system]
requires = ["poetry-core>=1.0.0"]
build-backend = "poetry.core.masonry.api"
[tool.poetry.scripts]
yoeo-detect = "yoeo.detect:run"
yoeo-train = "yoeo.train:run"
yoeo-test = "yoeo.test:run"
yoeo-to-onnx = "yoeo.scripts.convertPyTorchModelToONNX:run"
yoeo-onnx-to-openvino = "yoeo.scripts.convertONNXModelToOpenVinoIR:run"
yoeo-onnx-to-tvm = "yoeo.scripts.convertONNXModelToTVM:run"