Update development requirements and configuration for RT-DETRv2 integ…

…ration
supervisely-ecosystem · Dec 3, 2024 · 0739562 · 0739562
1 parent 618a15f
commit 0739562
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 7 deletions.
diff --git a/dev_requirements.txt b/dev_requirements.txt
@@ -1,3 +1,5 @@
+git+https://github.com/supervisely/supervisely.git@some-test-branch
+
 supervisely
 pycocotools
 torch==2.0.1
@@ -7,4 +9,5 @@ imgaug
 onnx==1.14.0
 onnxruntime==1.15.1
 PyYAML
-scipy
+scipy
+tensorboardX
diff --git a/requirements.txt b/requirements.txt
diff --git a/supervisely_integration/serve/config.json b/supervisely_integration/serve/config.json
@@ -11,7 +11,7 @@
 	"gpu": "preferred",
 	"session_tags": ["deployed_nn"],
 	"community_agent": false,
-	"docker_image": "supervisely/rt-detr:1.0.4",
+	"docker_image": "supervisely/rt-detrv2:1.0.0",
 	"entrypoint": "python3 -m uvicorn main:model.app --host 0.0.0.0 --port 8000 --ws websockets --app-dir supervisely_integration/serve",
 	"port": 8000,
 	"task_location": "application_sessions",

diff --git a/supervisely_integration/train/config.json b/supervisely_integration/train/config.json
@@ -9,7 +9,7 @@
 	"icon_cover": true,
 	"poster": "https://github.com/user-attachments/assets/ae6ec55e-63eb-43d1-99de-f62650939c69",
 	"description": "Train RT-DETR model on your data",
-	"docker_image": "supervisely/rt-detr:1.0.4",
+	"docker_image": "supervisely/rt-detrv2:1.0.0",
 	"instance_version": "6.11.22",
 	"task_location": "workspace_tasks",
 	"need_gpu": true,