Skip to content

Commit cf255cb

Browse files
authored
[Video-Ingestion] Peace of mind: adding default values for converted model path (#1390)
Signed-off-by: Krishna Murti <[email protected]>
1 parent a99fc58 commit cf255cb

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

sample-applications/video-search-and-summarization/chart/subchart/video-ingestion/templates/video-ingestion-configmap.yaml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -53,7 +53,7 @@ data:
5353
help="Name of the model (without extension, e.g., 'yolov8l-worldv2')")
5454
parser.add_argument("--model-type", type=str,
5555
help="Type of model (e.g., 'yolo_v8', 'YOLOv8-SEG')")
56-
parser.add_argument("--output-dir", type=str, default='ov_models/yoloworld',
56+
parser.add_argument("--output-dir", type=str, default='/models/yoloworld/v2',
5757
help="Directory to save the converted models")
5858
5959
args = parser.parse_args()
@@ -70,7 +70,7 @@ data:
7070
# Default model settings (can be overridden by environment variables)
7171
export OD_MODEL_NAME=${OD_MODEL_NAME}
7272
export OD_MODEL_TYPE=${OD_MODEL_TYPE}
73-
export OD_MODEL_OUTPUT_DIR=${OD_MODEL_OUTPUT_DIR:-"/home/pipeline-server/models/yoloworld"}
73+
export OD_MODEL_OUTPUT_DIR=${OD_MODEL_OUTPUT_DIR:-"/models/yoloworld/v2"}
7474
7575
# Function to convert object detection models
7676
convert_object_detection_models() {

sample-applications/video-search-and-summarization/video-ingestion/resources/scripts/converter.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -48,7 +48,7 @@ def main():
4848
help="Name of the model (without extension, e.g., 'yolov8l-worldv2')")
4949
parser.add_argument("--model-type", type=str, default='yolo_v8',
5050
help="Type of model (e.g., 'yolo_v8', 'YOLOv8-SEG')")
51-
parser.add_argument("--output-dir", type=str, default='ov_models/yoloworld',
51+
parser.add_argument("--output-dir", type=str, default='ov_models/yoloworld/v2',
5252
help="Directory to save the converted models")
5353

5454
args = parser.parse_args()

0 commit comments

Comments
 (0)