Using trt backend

peiva-git · Dec 30, 2023 · e8e659b · e8e659b
1 parent 177cf44
commit e8e659b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/basketballdetector/predicting/predict_ball_mask.py b/basketballdetector/predicting/predict_ball_mask.py
@@ -57,7 +57,7 @@ def __setup_model(self) -> fd.vision.segmentation.PaddleSegModel:
         option = fd.RuntimeOption()
         option.use_gpu()
         if self.__use_trt:
-            option.use_paddle_infer_backend()
+            option.use_trt_backend()
             option.paddle_infer_option.enable_trt = True
             option.trt_option.set_shape(
                 'x',