rename forward_test to predict

open-mmlab · Dec 21, 2022 · 09c4346 · 09c4346
1 parent acae7a7
commit 09c4346
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/mmdeploy/codebase/mmdet/deploy/object_detection_model.py b/mmdeploy/codebase/mmdet/deploy/object_detection_model.py
@@ -352,7 +352,7 @@ class labels of shape [N, num_det].
         ret = [r.cpu() for r in ret]
         return ret
 
-    def forward_test(self, imgs: Tensor, *args, **kwargs):
+    def predict(self, imgs: Tensor, *args, **kwargs):
         """Implement forward test.
 
         Args:
@@ -517,7 +517,7 @@ def partition1_postprocess(self, rois: Tensor, cls_score: Tensor,
             img_metas[0][0]['scale_factor'],
             cfg=rcnn_test_cfg)
 
-    def forward_test(self, imgs: Tensor, img_metas: Sequence[dict], *args,
+    def predict(self, imgs: Tensor, img_metas: Sequence[dict], *args,
                      **kwargs):
         """Implement forward test.