cleanup

neuralmagic · Jul 17, 2023 · 232de20 · 232de20
1 parent 2d3f6cd
commit 232de20
Showing 1 changed file with 0 additions and 7 deletions.
diff --git a/src/deepsparse/yolov8/pipelines.py b/src/deepsparse/yolov8/pipelines.py
@@ -83,20 +83,13 @@ def process_engine_outputs(
         """
         if self.subtask == "segmentation":
             if len(engine_outputs) != 2:
-                # print(engine_outputs)
                 warnings.warn(
                     "YOLOv8 Segmentation pipeline expects 2 outputs from engine, "
                     "got {}. Assuming first output is detection output, and last is segmentation output".format(
                         len(engine_outputs)
                     )
                 )
-                # engine_outputs = [engine_outputs[4], engine_outputs[0]]
                 engine_outputs = [engine_outputs[0], engine_outputs[5]]
-                # raise ValueError(
-                #     "Error: Segmentation pipeline "
-                #     "expects 2 outputs from engine"
-                #     "(detections and masks), got {}".format(len(engine_outputs))
-                # )
             return self.process_engine_outputs_seg(
                 engine_outputs=engine_outputs, **kwargs
             )