diff --git a/script/app-mlperf-inference/meta.yaml b/script/app-mlperf-inference/meta.yaml index 03787fdad..fb2dbd077 100644 --- a/script/app-mlperf-inference/meta.yaml +++ b/script/app-mlperf-inference/meta.yaml @@ -1720,7 +1720,7 @@ variations: - "yes" mounts: - "${{ MLC_ML_MODEL_YOLOV11_PATH }}:${{ MLC_ML_MODEL_YOLOV11_PATH }}" - - "${{ MLC_DATASET_YOLO_COCO2017_FILTERED_PATH }}:${{ MLC_DATASET_YOLO_COCO2017_FILTERED_PATH }}" + - "${{ MLC_ML_DATASET_MLPERF_INFERENCE_YOLO_COCO2017_FILTERED_DATASET_PATH }}:${{ MLC_ML_DATASET_MLPERF_INFERENCE_YOLO_COCO2017_FILTERED_DATASET_PATH }}" posthook_deps: - enable_if_env: MLC_MLPERF_LOADGEN_MODE: diff --git a/script/get-dataset-mlperf-inference-yolo-coco2017-filtered-dataset/customize.py b/script/get-dataset-mlperf-inference-yolo-coco2017-filtered-dataset/customize.py index 615fac7a9..e649b9a63 100644 --- a/script/get-dataset-mlperf-inference-yolo-coco2017-filtered-dataset/customize.py +++ b/script/get-dataset-mlperf-inference-yolo-coco2017-filtered-dataset/customize.py @@ -24,8 +24,9 @@ def postprocess(i): env = i['env'] if env.get('MLC_DOWNLOAD_MODE', '') != 'dry': - env['MLC_ML_DATASET_MLPERF_INFERENCE_YOLO_COCO2017_FILTERED_DATASET_PATH'] = os.path.join( - env['MLC_ML_DATASET_MLPERF_INFERENCE_YOLO_COCO2017_FILTERED_DATASET_PATH'], "val2017_safe") + if env.get('MLC_TMP_REQUIRE_DOWNLOAD', '') == "yes": + env['MLC_ML_DATASET_MLPERF_INFERENCE_YOLO_COCO2017_FILTERED_DATASET_PATH'] = os.path.join( + env['MLC_ML_DATASET_MLPERF_INFERENCE_YOLO_COCO2017_FILTERED_DATASET_PATH'], "val2017_safe") env['MLC_ML_DATASET_FILE_WITH_PATH'] = env['MLC_ML_DATASET_MLPERF_INFERENCE_YOLO_COCO2017_FILTERED_DATASET_PATH'] env['MLC_ML_DATASET_MLPERF_INFERENCE_YOLO_COCO2017_FILTERED_DATASET_ANNOTATION_PATH'] = os.path.join( env['MLC_ML_DATASET_FILE_WITH_PATH'], "annotations", "instances_val2017.json") diff --git a/script/get-ml-model-yolov11/customize.py b/script/get-ml-model-yolov11/customize.py index 9561190e3..0559f3620 100644 --- a/script/get-ml-model-yolov11/customize.py +++ b/script/get-ml-model-yolov11/customize.py @@ -23,8 +23,9 @@ def postprocess(i): env = i['env'] if env.get('MLC_DOWNLOAD_MODE', '') != 'dry': - env['MLC_ML_MODEL_YOLOV11_PATH'] = os.path.join( - env['MLC_ML_MODEL_YOLOV11_PATH'], 'yolo11l.pt') + if env.get('MLC_TMP_REQUIRE_DOWNLOAD') == "yes": + env['MLC_ML_MODEL_YOLOV11_PATH'] = os.path.join( + env['MLC_ML_MODEL_YOLOV11_PATH'], 'yolo11l.pt') env['MLC_ML_MODEL_FILE_WITH_PATH'] = env['MLC_ML_MODEL_YOLOV11_PATH'] return {'return': 0}