From 07395620f0334aefe7e3e7fac503f2415eef01cf Mon Sep 17 00:00:00 2001 From: cxnt Date: Tue, 3 Dec 2024 16:40:33 +0400 Subject: [PATCH] Update development requirements and configuration for RT-DETRv2 integration --- dev_requirements.txt | 5 ++++- requirements.txt | 4 ---- supervisely_integration/serve/config.json | 2 +- supervisely_integration/train/config.json | 2 +- 4 files changed, 6 insertions(+), 7 deletions(-) delete mode 100644 requirements.txt diff --git a/dev_requirements.txt b/dev_requirements.txt index 520d79d50..5ec2ffbbf 100644 --- 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 \ No newline at end of file +scipy +tensorboardX diff --git a/requirements.txt b/requirements.txt deleted file mode 100644 index 699140a2e..000000000 --- a/requirements.txt +++ /dev/null @@ -1,4 +0,0 @@ -git+https://github.com/supervisely/supervisely.git@serving-gui-template - -tensorboard -tensorboardX \ No newline at end of file diff --git a/supervisely_integration/serve/config.json b/supervisely_integration/serve/config.json index 5b87c5169..6d9f937cf 100644 --- 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 index e766de405..9794e6c9a 100644 --- 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,