Added command to untar waymo dataset files (#274)

* Added command to untar waymo dataset files * Clean run-mobilenet-models code
mlcommons · Feb 25, 2025 · 72fbb7a · 72fbb7a
1 parent 231a219
commit 72fbb7a
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/script/get-dataset-waymo/run-rclone.sh b/script/get-dataset-waymo/run-rclone.sh
@@ -1,4 +1,7 @@
 cmd="rclone sync mlc-waymo:waymo_preprocessed_dataset/kitti_format ${MLC_DATASET_WAYMO_PATH} -P"
 echo $cmd
 eval $cmd
-test $? -eq 0 || exit $?
+test $? -eq 0 || exit $?
+cd ${MLC_DATASET_WAYMO_PATH}/kitti_format/training
+for f in *.tar.gz; do tar -xzvf "$f"; done
+cd -
diff --git a/script/run-mlperf-inference-mobilenet-models/customize.py b/script/run-mlperf-inference-mobilenet-models/customize.py
@@ -200,6 +200,8 @@ def preprocess(i):
                     if r['return'] > 0:
                         print(r)
                     #    return r
+                    else:
+                        importlib.reload(mlc.action)
 
                 if is_true(env.get('MLC_TEST_ONE_RUN', '')):
                     return {'return': 0}