-
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.5.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.8,<4.0"
numpy = "^1.21.1"
torch = ">=1.10.1, <1.13.0"
torchvision = ">=0.8.2"
matplotlib = "^3.3.3"
tensorboard = "^2.12.2"
terminaltables = "^3.1.10"
Pillow = "^9.1.0"
tqdm = "^4.64.1"
imgaug = "^0.4.0"
torchsummary = "^1.5.1"
PyYAML = "^6.0"
opencv-python = "^4.5.2"
[tool.poetry.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"