working pipeline with ColonSegNet_brightcontr.onnx but maybe more fra…

…mes are required for test video #18
SciKit-Surgery · Nov 10, 2023 · f91306f · f91306f
1 parent 9b7789b
commit f91306f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/nvidia-clara-agx/colonoscopy_segmentation/colonoscopy_segmentation.py b/nvidia-clara-agx/colonoscopy_segmentation/colonoscopy_segmentation.py
@@ -54,8 +54,9 @@ def __init__(self, data, source="replayer"):
         self.sample_data_path = data
 
         self.model_path_map = {
-            "ultrasound_seg": os.path.join(self.sample_data_path, "ColonSegNet-07112023-2359.onnx"),
+            #"ultrasound_seg": os.path.join(self.sample_data_path, "ColonSegNet-07112023-2359.onnx"),
             #"ultrasound_seg": os.path.join(self.sample_data_path, "colon.onnx"),
+            "ultrasound_seg": os.path.join(self.sample_data_path, "ColonSegNet_brightcontr.onnx"),
         }
 
     def compose(self):