56 lines
1.9 KiB
TOML
56 lines
1.9 KiB
TOML
[tool.poetry]
|
|
name = "Trajectron-plus-plus"
|
|
version = "0.1.1"
|
|
description = "This repository contains the code for Trajectron++: Dynamically-Feasible Trajectory Forecasting With Heterogeneous Data by Tim Salzmann*, Boris Ivanovic*, Punarjay Chakravarty, and Marco Pavone (* denotes equal contribution)."
|
|
authors = ["Ruben van de Ven <git@rubenvandeven.com>"]
|
|
readme = "README.md"
|
|
license = "MIT"
|
|
|
|
packages = [
|
|
{ include = "trajectron" },
|
|
]
|
|
|
|
[tool.poetry.scripts]
|
|
trajectron_train = "trajectron.train:main"
|
|
|
|
[tool.poetry.dependencies]
|
|
python = "^3.9,<3.12"
|
|
numpy = "^1.24.3"
|
|
opencv-python = "^4.7.0.72"
|
|
ipykernel = "^6.22.0"
|
|
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'" },
|
|
#]
|
|
|
|
#av = "^10.0.0"
|
|
matplotlib = "^3.5"
|
|
#numba = "^0.57.0"
|
|
#scikit-image = "^0.20.0"
|
|
#scikit-learn = "0.22.1"
|
|
#filterpy = "^1.4.5"
|
|
tqdm = "^4.65.0"
|
|
scipy = "^1.11.3"
|
|
pandas = "^2.1.1"
|
|
orjson = "^3.9.7"
|
|
nuscenes-devkit = "^1.1.11"
|
|
pyquaternion = "^0.9.9"
|
|
dill = "^0.3.7"
|
|
ncls = "^0.0.68"
|
|
notebook = "^7.0.4"
|
|
scikit-learn = "^1.3.1"
|
|
seaborn = "^0.13.0"
|
|
setuptools = "^68.2.2"
|
|
tensorboard = "^2.11"
|
|
tensorboardx = "^2.6.2.2"
|
|
|
|
|
|
[build-system]
|
|
requires = ["poetry-core"]
|
|
build-backend = "poetry.core.masonry.api"
|