50 lines
2 KiB
TOML
50 lines
2 KiB
TOML
[tool.poetry]
|
|
name = "trap"
|
|
version = "0.1.0"
|
|
description = "Art installation with trajectory prediction"
|
|
authors = ["Ruben van de Ven <git@rubenvandeven.com>"]
|
|
readme = "README.md"
|
|
|
|
[tool.poetry.scripts]
|
|
trapserv = "trap.plumber:start"
|
|
tracker = "trap.tools:tracker_preprocess"
|
|
compare = "trap.tools:tracker_compare"
|
|
process_data = "trap.process_data:main"
|
|
blacklist = "trap.tools:blacklist_tracks"
|
|
rewrite_tracks = "trap.tools:rewrite_raw_track_files"
|
|
|
|
|
|
[tool.poetry.dependencies]
|
|
python = "^3.10,<3.12,"
|
|
|
|
#trajectron-plus-plus = { path = "../Trajectron-plus-plus/", develop = true }
|
|
trajectron-plus-plus = { git = "https://git.rubenvandeven.com/security_vision/Trajectron-plus-plus/" }
|
|
torch = [
|
|
{ version="1.12.1" },
|
|
# { url = "https://download.pytorch.org/whl/cu113/torch-1.12.1%2Bcu113-cp38-cp38-linux_x86_64.whl", markers = "python_version ~= '3.8' and sys_platform == 'linux'" },
|
|
{ url = "https://download.pytorch.org/whl/cu113/torch-1.12.1%2Bcu113-cp310-cp310-linux_x86_64.whl", markers = "python_version ~= '3.10' and sys_platform == 'linux'" },
|
|
]
|
|
|
|
torchvision = [
|
|
{ version="0.13.1" },
|
|
# { url = "https://download.pytorch.org/whl/cu113/torchvision-0.13.1%2Bcu113-cp38-cp38-linux_x86_64.whl", markers = "python_version ~= '3.8' and sys_platform == 'linux'" },
|
|
{ url = "https://download.pytorch.org/whl/cu113/torchvision-0.13.1%2Bcu113-cp310-cp310-linux_x86_64.whl", markers = "python_version ~= '3.10' and sys_platform == 'linux'" },
|
|
]
|
|
deep-sort-realtime = "^1.3.2"
|
|
ultralytics = "^8.3"
|
|
ffmpeg-python = "^0.2.0"
|
|
torchreid = "^0.2.5"
|
|
gdown = "^4.7.1"
|
|
pandas-helper-calc = {git = "https://github.com/scls19fr/pandas-helper-calc"}
|
|
tsmoothie = "^1.0.5"
|
|
pyglet = "^2.0.15"
|
|
pyglet-cornerpin = "^0.3.0"
|
|
opencv-python = {file="./opencv_python-4.10.0.84-cp310-cp310-linux_x86_64.whl"}
|
|
setproctitle = "^1.3.3"
|
|
bytetracker = { git = "https://github.com/rubenvandeven/bytetrack-pip" }
|
|
jsonlines = "^4.0.0"
|
|
tensorboardx = "^2.6.2.2"
|
|
|
|
[build-system]
|
|
requires = ["poetry-core"]
|
|
build-backend = "poetry.core.masonry.api"
|