Add dummy input for VPT

src/otx/core/model/anomaly.py

@@ -401,4 +401,4 @@ def get_dummy_input(self, batch_size: int = 1) -> AnomalyModelInputs:
                 labels=[torch.LongTensor(0)],
                 boxes=torch.tensor(0),
                 masks=torch.tensor(0),
-            )
+            )

src/otx/core/model/visual_prompting.py

@@ -255,6 +255,14 @@ def _set_label_info(self, _: LabelInfoTypes) -> None:
         msg = f"Reconfiguring label_info has no effect on {self.__class__.__name__}."
         log.warning(msg)
 
+    def get_dummy_input(self, batch_size: int = 1) -> VisualPromptingBatchDataEntity:
+        """Returns a dummy input for VPT model"""
+        images = [torch.rand(3, self.model.image_size, self.model.image_size) for _ in range(batch_size)]
+        labels = [{"points": torch.LongTensor([0] * batch_size)}] * batch_size
+        prompts = [torch.zeros((1, 2))] * batch_size
+        return VisualPromptingBatchDataEntity(batch_size, images, imgs_info=[], labels=labels, points=prompts,
+                                              masks=[None]*batch_size, polygons=[None]*batch_size, bboxes=[None]*batch_size)
+
 
 class OTXZeroShotVisualPromptingModel(
     OTXModel[ZeroShotVisualPromptingBatchDataEntity, ZeroShotVisualPromptingBatchPredEntity],
@@ -421,6 +429,20 @@ def _set_label_info(self, _: LabelInfoTypes) -> None:
         msg = f"Reconfiguring label_info has no effect on {self.__class__.__name__}."
         log.warning(msg)
 
+    def get_dummy_input(self, batch_size: int = 1) -> ZeroShotVisualPromptingBatchDataEntity:
+        """Returns a dummy input for ZSL VPT model"""
+        images = [torch.rand(3, self.model.image_size, self.model.image_size) for _ in range(batch_size)]
+        labels = [{"points": torch.LongTensor([0] * batch_size)}] * batch_size
+        prompts = [torch.zeros((1, 2))] * batch_size
+        infos = []
+        for i, img in enumerate(images):
+            infos.append(ImageInfo(
+                img_idx=i,
+                img_shape=img.shape,
+                ori_shape=img.shape,
+            ))
+        return ZeroShotVisualPromptingBatchDataEntity(batch_size, images, imgs_info=infos, labels=labels, prompts=prompts, masks=[], polygons=[])
+
 
 class OVVisualPromptingModel(
     OVModel[
@@ -1340,4 +1362,4 @@ def get_dummy_input(self, batch_size: int = 1) -> ZeroShotVisualPromptingBatchDa
                 img_shape=img.shape,
                 ori_shape=img.shape,
             ))
-        return ZeroShotVisualPromptingBatchDataEntity(batch_size, images, imgs_info=infos, labels=labels, prompts=prompts, masks=[], polygons=[])
+        return ZeroShotVisualPromptingBatchDataEntity(batch_size, images, imgs_info=infos, labels=labels, prompts=prompts, masks=[], polygons=[])