pritmanvar-bacancy commited on
Commit
b16bd7f
·
verified ·
1 Parent(s): 3367983

Update pipeline_functions.py

Browse files
Files changed (1) hide show
  1. pipeline_functions.py +2 -0
pipeline_functions.py CHANGED
@@ -19,6 +19,8 @@ path = {
19
  'MAIN_FLOW_GRAY_IMG_DIR_PATH': str(os.getenv('MAIN_FLOW_GRAY_IMG_DIR_PATH')),
20
  'MAIN_FLOW_INFERENCE_FOLDER': str(os.getenv('MAIN_FLOW_INFERENCE_FOLDER')),
21
  }
 
 
22
  seg_model = YOLO(path['SEG_MODEL_PATH'])
23
 
24
  CONF = 0.7
 
19
  'MAIN_FLOW_GRAY_IMG_DIR_PATH': str(os.getenv('MAIN_FLOW_GRAY_IMG_DIR_PATH')),
20
  'MAIN_FLOW_INFERENCE_FOLDER': str(os.getenv('MAIN_FLOW_INFERENCE_FOLDER')),
21
  }
22
+
23
+ print(path['SEG_MODEL_PATH'])
24
  seg_model = YOLO(path['SEG_MODEL_PATH'])
25
 
26
  CONF = 0.7