From ab1bd1ec40bbbcbf3c5fa8e810cc8eb45f92066f Mon Sep 17 00:00:00 2001
From: Pavel Bartsits <48913536+cxnt@users.noreply.github.com>
Date: Fri, 13 Dec 2024 19:28:29 +0400
Subject: [PATCH] Update docker image version (#2)
* Create requirements.txt
* Add demo usage instructions to app_options.yaml
* Update screenshots in README
* Update requirements.txt
* Update Dockerfile and configuration for RT-DETRv2 to version 1.0.3; remove requirements.txt
---
docker/Dockerfile | 2 +-
docker/publish.sh | 4 ++--
supervisely_integration/serve/config.json | 2 +-
supervisely_integration/train/README.md | 16 ++++++++--------
supervisely_integration/train/app_options.yaml | 7 +++++++
supervisely_integration/train/config.json | 2 +-
6 files changed, 20 insertions(+), 13 deletions(-)
diff --git a/docker/Dockerfile b/docker/Dockerfile
index afb67315..be634252 100644
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -31,7 +31,7 @@ RUN pip install --no-cache-dir \
tensorboard \
imgaug==0.4.0
-RUN pip install --no-cache-dir supervisely[training]==6.73.251
+RUN pip install --no-cache-dir supervisely[training]==6.73.255
RUN mkdir -p ~/.cache/supervisely/checkpoints
RUN curl -L -o ~/.cache/supervisely/checkpoints/rtdetrv2_r18vd_120e_coco_rerun_48.1.pth https://github.com/lyuwenyu/storage/releases/download/v0.2/rtdetrv2_r18vd_120e_coco_rerun_48.1.pth
diff --git a/docker/publish.sh b/docker/publish.sh
index 97170231..2dbb9907 100644
--- a/docker/publish.sh
+++ b/docker/publish.sh
@@ -1,2 +1,2 @@
-docker build -t supervisely/rt-detrv2:1.0.2 . && \
-docker push supervisely/rt-detrv2:1.0.2
+docker build -t supervisely/rt-detrv2:1.0.3 . && \
+docker push supervisely/rt-detrv2:1.0.3
diff --git a/supervisely_integration/serve/config.json b/supervisely_integration/serve/config.json
index 309a837e..746703da 100644
--- a/supervisely_integration/serve/config.json
+++ b/supervisely_integration/serve/config.json
@@ -13,7 +13,7 @@
"gpu": "preferred",
"session_tags": ["deployed_nn"],
"community_agent": false,
- "docker_image": "supervisely/rt-detrv2:1.0.2",
+ "docker_image": "supervisely/rt-detrv2:1.0.3",
"task_location": "application_sessions",
"license": {
"type": "Apache-2.0"
diff --git a/supervisely_integration/train/README.md b/supervisely_integration/train/README.md
index c7cde603..226db674 100644
--- a/supervisely_integration/train/README.md
+++ b/supervisely_integration/train/README.md
@@ -37,31 +37,31 @@ Train RT-DETRv2 models in Supervisely on your custom data. RT-DETRv2 (Real-Time
**Step 1.** Select if you want to use cached project or redownload it
-
+
**Step 2.** Select train / val split
-
+
**Step 3.** Select the classes you want to train RT-DETRv2 on
-
+
**Step 4.** Select the model you want to train
-
+
**Step 5.** Configure hyperaparameters and select whether you want to use model evaluation and convert checkpoints to ONNX and TensorRT
-
+
**Step 6.** Enter experiment name and start training
-
+
**Step 7.** Monitor training progress
-
+
# Obtain saved checkpoints
@@ -69,7 +69,7 @@ All the trained checkpoints, that are generated through the process of training
You will see a folder thumbnail with a link to your saved checkpoints by the end of training process.
-
+
# Acknowledgment
diff --git a/supervisely_integration/train/app_options.yaml b/supervisely_integration/train/app_options.yaml
index 12108cac..c5b027be 100644
--- a/supervisely_integration/train/app_options.yaml
+++ b/supervisely_integration/train/app_options.yaml
@@ -8,6 +8,13 @@ model_benchmark: true
export_onnx_supported: true
export_tensorrt_supported: true
+# How to use models outside of the Supervisely platform
+demo:
+ overview: "https://github.com/supervisely-ecosystem/RT-DETRv2/blob/main/supervisely_integration/demo/README.md"
+ pytorch: "https://github.com/supervisely-ecosystem/RT-DETRv2/blob/main/supervisely_integration/demo/demo_pytorch.py"
+ onnx: "https://github.com/supervisely-ecosystem/RT-DETRv2/blob/main/supervisely_integration/demo/demo_onnx.py"
+ tensorrt: "https://github.com/supervisely-ecosystem/RT-DETRv2/blob/main/supervisely_integration/demo/demo_tensorrt.py"
+
# Enable this option when using supervisely train logger
# train_logger: "tensorboard"
diff --git a/supervisely_integration/train/config.json b/supervisely_integration/train/config.json
index b436bff0..95c75954 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/b4554de5-3d2c-4b4f-aba9-95864cb05289",
"description": "Train RT-DETRv2 model on your data",
- "docker_image": "supervisely/rt-detrv2:1.0.2",
+ "docker_image": "supervisely/rt-detrv2:1.0.3",
"instance_version": "6.12.12",
"task_location": "workspace_tasks",
"need_gpu": true,