Migrate to uv

README.md

@@ -3,7 +3,7 @@
 ## Install
 
 * Run `bash build_opencv_with_gstreamer.sh` to build opencv with gstreamer support
-* Use pyenv + poetry to install
+* Use `uv` to install
 
 ## How to
 

pyproject.toml

@@ -1,11 +1,36 @@
-[tool.poetry]
+[project]
 name = "trap"
 version = "0.1.0"
 description = "Art installation with trajectory prediction"
-authors = ["Ruben van de Ven <git@rubenvandeven.com>"]
+authors = [{ name = "Ruben van de Ven", email = "git@rubenvandeven.com" }]
+requires-python = "~=3.10.4"
 readme = "README.md"
+dependencies = [
+    "trajectron-plus-plus",
+    "torch==1.12.1",
+    "torchvision==0.13.1",
+    "deep-sort-realtime>=1.3.2,<2",
+    "ultralytics~=8.3",
+    "ffmpeg-python>=0.2.0,<0.3",
+    "torchreid>=0.2.5,<0.3",
+    "gdown>=4.7.1,<5",
+    "pandas-helper-calc",
+    "tsmoothie>=1.0.5,<2",
+    "pyglet>=2.0.15,<3",
+    "pyglet-cornerpin>=0.3.0,<0.4",
+    "opencv-python",
+    "setproctitle>=1.3.3,<2",
+    "bytetracker",
+    "jsonlines>=4.0.0,<5",
+    "tensorboardx>=2.6.2.2,<3",
+    "shapely>=1,<2",
+    "baumer-neoapi",
+    "qrcode~=8.0",
+    "pyusb>=1.3.1,<2",
+    "ipywidgets>=8.1.5,<9",
+]
 
-[tool.poetry.scripts]
+[project.scripts]
 trapserv = "trap.plumber:start"
 tracker = "trap.tools:tracker_preprocess"
 compare = "trap.tools:tracker_compare"
@@ -13,42 +38,16 @@ 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,"
+[tool.uv]
 
-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"
+[tool.uv.sources]
+trajectron-plus-plus = { path = "../Trajectron-plus-plus/", editable = true }
+torch = [{ url = "https://download.pytorch.org/whl/cu113/torch-1.12.1%2Bcu113-cp310-cp310-linux_x86_64.whl", marker = "python_version ~= '3.10' and sys_platform == 'linux'" }]
+torchvision = [{ url = "https://download.pytorch.org/whl/cu113/torchvision-0.13.1%2Bcu113-cp310-cp310-linux_x86_64.whl", marker = "python_version ~= '3.10' and sys_platform == 'linux'" }]
+pandas-helper-calc = { git = "https://github.com/scls19fr/pandas-helper-calc" }
 bytetracker = { git = "https://github.com/rubenvandeven/bytetrack-pip" }
-jsonlines = "^4.0.0"
-tensorboardx = "^2.6.2.2"
-shapely = "^1"
-baumer-neoapi = {path = "../../Downloads/Baumer_neoAPI_1.4.1_lin_x86_64_python/wheel/baumer_neoapi-1.4.1-cp34.cp35.cp36.cp37.cp38.cp39.cp310.cp311.cp312-none-linux_x86_64.whl"}
-qrcode = "^8.0"
-pyusb = "^1.3.1"
-ipywidgets = "^8.1.5"
+baumer-neoapi = { path = "../../Downloads/Baumer_neoAPI_1.4.1_lin_x86_64_python/wheel/baumer_neoapi-1.4.1-cp34.cp35.cp36.cp37.cp38.cp39.cp310.cp311.cp312-none-linux_x86_64.whl" }
 
 [build-system]
-requires = ["poetry-core"]
-build-backend = "poetry.core.masonry.api"
+requires = ["hatchling"]
+build-backend = "hatchling.build"