diff --git a/script/get-dataset-waymo/COPYRIGHT.md b/script/get-dataset-waymo/COPYRIGHT.md new file mode 100644 index 000000000..d2ceead84 --- /dev/null +++ b/script/get-dataset-waymo/COPYRIGHT.md @@ -0,0 +1,9 @@ +# Copyright Notice + +© 2025-2026 MLCommons. All Rights Reserved. + +This file is licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance with the License. A copy of the License can be obtained at: + +[Apache License, Version 2.0](http://www.apache.org/licenses/LICENSE-2.0) + +Unless required by applicable law or agreed to in writing, software distributed under the License is provided on an "AS IS" basis, without warranties or conditions of any kind, either express or implied. Please refer to the License for the specific language governing permissions and limitations under the License. diff --git a/script/get-dataset-waymo/customize.py b/script/get-dataset-waymo/customize.py new file mode 100644 index 000000000..273feef06 --- /dev/null +++ b/script/get-dataset-waymo/customize.py @@ -0,0 +1,28 @@ +from mlc import utils +import os + + +def preprocess(i): + + os_info = i['os_info'] + + env = i['env'] + + if os_info['platform'] == "windows": + return {'return': 1, 'error': 'Script not supported in windows yet!'} + + if env.get('MLC_DATASET_WAYMO_PATH', '') == '': + return {'return': 1, 'error': 'Please provide path to kitti dataset using tag \\`--waymo_path\\`as automatic download of this dataset is not supported yet.'} + + if not os.path.exists(env['MLC_DATASET_WAYMO_PATH']): + return { + 'return': 1, 'error': f"Path {env['MLC_DATASET_WAYMO_PATH']} does not exists!"} + + return {'return': 0} + + +def postprocess(i): + + env = i['env'] + + return {'return': 0} diff --git a/script/get-dataset-waymo/meta.yaml b/script/get-dataset-waymo/meta.yaml new file mode 100644 index 000000000..bfbba995f --- /dev/null +++ b/script/get-dataset-waymo/meta.yaml @@ -0,0 +1,19 @@ +alias: get-dataset-waymo +automation_alias: script +automation_uid: 5b4e0237da074764 +cache: true +tags: +- get +- dataset +- waymo +uid: 21b269c753b64437 +new_env_keys: + - MLC_DATASET_WAYMO_PATH +input_mapping: + waymo_path: MLC_DATASET_WAYMO_PATH +variations: + kitti_format: + default: true + group: dataset-format + env: + MLC_DATASET_WAYMO_FORMAT: kitti diff --git a/script/get-dataset-waymo/run.sh b/script/get-dataset-waymo/run.sh new file mode 100644 index 000000000..3197bb8ad --- /dev/null +++ b/script/get-dataset-waymo/run.sh @@ -0,0 +1,8 @@ +#!/bin/bash + +#CM Script location: ${MLC_TMP_CURRENT_SCRIPT_PATH} + +#To export any variable +#echo "VARIABLE_NAME=VARIABLE_VALUE" >>tmp-run-env.out + +#${MLC_PYTHON_BIN_WITH_PATH} contains the path to python binary if "get,python" is added as a dependency diff --git a/script/get-ml-model-pointpillars/COPYRIGHT.md b/script/get-ml-model-pointpillars/COPYRIGHT.md new file mode 100644 index 000000000..d2ceead84 --- /dev/null +++ b/script/get-ml-model-pointpillars/COPYRIGHT.md @@ -0,0 +1,9 @@ +# Copyright Notice + +© 2025-2026 MLCommons. All Rights Reserved. + +This file is licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance with the License. A copy of the License can be obtained at: + +[Apache License, Version 2.0](http://www.apache.org/licenses/LICENSE-2.0) + +Unless required by applicable law or agreed to in writing, software distributed under the License is provided on an "AS IS" basis, without warranties or conditions of any kind, either express or implied. Please refer to the License for the specific language governing permissions and limitations under the License. diff --git a/script/get-ml-model-pointpillars/customize.py b/script/get-ml-model-pointpillars/customize.py new file mode 100644 index 000000000..b6685c889 --- /dev/null +++ b/script/get-ml-model-pointpillars/customize.py @@ -0,0 +1,32 @@ +from mlc import utils +import os + + +def preprocess(i): + + os_info = i['os_info'] + + env = i['env'] + + if os_info['platform'] == "windows": + return {'return': 1, 'error': 'Script not supported in windows yet!'} + + if env.get('MLC_ML_MODEL_POINT_PILLARS_PATH', '') == '': + return {'return': 1, 'error': 'Please provide path to pointpillars model using tag \\`--pp_path\\`as automatic download of this model is not supported yet.'} + + if os.path.isdir(env['MLC_ML_MODEL_POINT_PILLARS_PATH']): + if env['MLC_ML_MODEL_PP_FORMAT'] == "onnx": + env['MLC_ML_MODEL_POINT_PILLARS_PATH'] = os.path.join( + env['MLC_ML_MODEL_POINT_PILLARS_PATH'], "pp.onnx") + else: + env['MLC_ML_MODEL_POINT_PILLARS_PATH'] = os.path.join( + env['MLC_ML_MODEL_POINT_PILLARS_PATH'], "pp_ep36.pth") + + return {'return': 0} + + +def postprocess(i): + + env = i['env'] + + return {'return': 0} diff --git a/script/get-ml-model-pointpillars/meta.yaml b/script/get-ml-model-pointpillars/meta.yaml new file mode 100644 index 000000000..18470e4c0 --- /dev/null +++ b/script/get-ml-model-pointpillars/meta.yaml @@ -0,0 +1,26 @@ +alias: get-ml-model-pointpillars +automation_alias: script +automation_uid: 5b4e0237da074764 +cache: true +tags: +- get +- ml-model +- ml +- model +- pointpillars +uid: 3562621a8994411d +new_env_keys: + - MLC_ML_MODEL_POINT_PILLARS_PATH +input_mapping: + pp_path: MLC_ML_MODEL_POINT_PILLARS_PATH +variations: + gpu: + default: true + group: device + env: + MLC_ML_MODEL_PP_FORMAT: pth + cpu: + group: device + env: + MLC_ML_MODEL_PP_FORMAT: onnx + diff --git a/script/get-ml-model-pointpillars/run.sh b/script/get-ml-model-pointpillars/run.sh new file mode 100644 index 000000000..3197bb8ad --- /dev/null +++ b/script/get-ml-model-pointpillars/run.sh @@ -0,0 +1,8 @@ +#!/bin/bash + +#CM Script location: ${MLC_TMP_CURRENT_SCRIPT_PATH} + +#To export any variable +#echo "VARIABLE_NAME=VARIABLE_VALUE" >>tmp-run-env.out + +#${MLC_PYTHON_BIN_WITH_PATH} contains the path to python binary if "get,python" is added as a dependency diff --git a/script/get-ml-model-resnet50-deeplab/COPYRIGHT.md b/script/get-ml-model-resnet50-deeplab/COPYRIGHT.md new file mode 100644 index 000000000..d2ceead84 --- /dev/null +++ b/script/get-ml-model-resnet50-deeplab/COPYRIGHT.md @@ -0,0 +1,9 @@ +# Copyright Notice + +© 2025-2026 MLCommons. All Rights Reserved. + +This file is licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance with the License. A copy of the License can be obtained at: + +[Apache License, Version 2.0](http://www.apache.org/licenses/LICENSE-2.0) + +Unless required by applicable law or agreed to in writing, software distributed under the License is provided on an "AS IS" basis, without warranties or conditions of any kind, either express or implied. Please refer to the License for the specific language governing permissions and limitations under the License. diff --git a/script/get-ml-model-resnet50-deeplab/customize.py b/script/get-ml-model-resnet50-deeplab/customize.py new file mode 100644 index 000000000..0df3b1c3f --- /dev/null +++ b/script/get-ml-model-resnet50-deeplab/customize.py @@ -0,0 +1,33 @@ +from mlc import utils +import os + + +def preprocess(i): + + os_info = i['os_info'] + + env = i['env'] + + if os_info['platform'] == "windows": + return {'return': 1, 'error': 'Script not supported in windows yet!'} + + if env.get('MLC_ML_MODEL_DPLAB_RESNET50_PATH', '') == '': + return {'return': 1, 'error': 'Please provide path to deeplab resnet 50 model using tag \\`--dp_resnet50_path\\`as automatic download of this dataset is not supported yet.'} + + if os.path.isdir(env['MLC_ML_MODEL_DPLAB_RESNET50_PATH']): + if env['MLC_ML_MODEL_DPLAB_RESNET50_FORMAT'] == "onnx": + env['MLC_ML_MODEL_DPLAB_RESNET50_PATH'] = os.path.join( + env['MLC_ML_MODEL_DPLAB_RESNET50_PATH'], "deeplabv3+.onnx") + else: + env['MLC_ML_MODEL_DPLAB_RESNET50_PATH'] = os.path.join( + env['MLC_ML_MODEL_DPLAB_RESNET50_PATH'], + "best_deeplabv3plus_resnet50_waymo_os16.pth") + + return {'return': 0} + + +def postprocess(i): + + env = i['env'] + + return {'return': 0} diff --git a/script/get-ml-model-resnet50-deeplab/meta.yaml b/script/get-ml-model-resnet50-deeplab/meta.yaml new file mode 100644 index 000000000..c8c8b84e1 --- /dev/null +++ b/script/get-ml-model-resnet50-deeplab/meta.yaml @@ -0,0 +1,27 @@ +alias: get-dataset-deeplab-resnet50 +automation_alias: script +automation_uid: 5b4e0237da074764 +cache: true +tags: +- get +- ml-model +- ml +- model +- resnet50-deeplab +- resnet50 +- deeplab +uid: 93097b691a6a4fce +new_env_keys: + - MLC_ML_MODEL_DPLAB_RESNET50_PATH +input_mapping: + dp_resnet50_path: MLC_ML_MODEL_DPLAB_RESNET50_PATH +variations: + gpu: + default: true + group: device + env: + MLC_ML_MODEL_DPLAB_RESNET50_FORMAT: pth + cpu: + group: device + env: + MLC_ML_MODEL_DPLAB_RESNET50_FORMAT: onnx diff --git a/script/get-ml-model-resnet50-deeplab/run.sh b/script/get-ml-model-resnet50-deeplab/run.sh new file mode 100644 index 000000000..3197bb8ad --- /dev/null +++ b/script/get-ml-model-resnet50-deeplab/run.sh @@ -0,0 +1,8 @@ +#!/bin/bash + +#CM Script location: ${MLC_TMP_CURRENT_SCRIPT_PATH} + +#To export any variable +#echo "VARIABLE_NAME=VARIABLE_VALUE" >>tmp-run-env.out + +#${MLC_PYTHON_BIN_WITH_PATH} contains the path to python binary if "get,python" is added as a dependency