diff --git a/dev_requirements.txt b/dev_requirements.txt index 520d79d5..5ec2ffbb 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 699140a2..00000000 --- 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 5b87c516..6d9f937c 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 e766de40..9794e6c9 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,