Update pipeline_functions.py
Browse files- 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
|