Update project configuration and inference settings; modify instance …

…version and add requirements for training

.vscode/launch.json

@@ -25,7 +25,7 @@
 				// "APP_NAME": "Train RT-DETR",
 				"TEAM_ID": "8",
 				"WORKSPACE_ID": "349",
-				"PROJECT_ID": "42201"
+				"PROJECT_ID": "43192"
 			}
 		},
 		{

requirements.txt

@@ -0,0 +1 @@
+git+https://github.com/supervisely/supervisely.git@train-app-progress

supervisely_integration/serve/config.json

@@ -18,5 +18,5 @@
 	"license": {
 		"type": "Apache-2.0"
 	},
-	"instance_version": "6.11.22"
+	"instance_version": "6.12.12"
 }

supervisely_integration/serve/rtdetrv2.py

@@ -90,7 +90,8 @@ def _predict_pytorch(
         with Timer() as preprocess_timer:
             imgs_pil = [Image.fromarray(img) for img in images_np]
             orig_target_sizes = torch.as_tensor([img.size for img in imgs_pil]).to(self.device)
-            samples = torch.stack(self.transforms(imgs_pil)).to(self.device)
+            transformed_imgs = [self.transforms(img) for img in imgs_pil]
+            samples = torch.stack(transformed_imgs).to(self.device)
         # 2. Inference
         with Timer() as inference_timer:
             outputs = self.model(samples)

supervisely_integration/train/config.json

@@ -10,7 +10,7 @@
 	"poster": "https://github.com/user-attachments/assets/ae6ec55e-63eb-43d1-99de-f62650939c69",
 	"description": "Train RT-DETR model on your data",
 	"docker_image": "supervisely/rt-detrv2:1.0.0",
-	"instance_version": "6.11.22",
+	"instance_version": "6.12.12",
 	"task_location": "workspace_tasks",
 	"need_gpu": true,
 	"gpu": "required",

supervisely_integration/train/main.py

@@ -23,7 +23,7 @@
     f"{base_path}/app_options.yaml",
 )
 
-inference_settings = {"confidence_threshold": 0.4}
+inference_settings = {"confidence_threshold": 0}
 train.register_inference_class(RTDETRv2Benchmark, inference_settings)
 
 # For debug