From 3b21f869aa3d3d2f87d77d7c54a213ef91921fb8 Mon Sep 17 00:00:00 2001 From: Nate Meyer Date: Thu, 6 Jul 2023 00:25:31 -0400 Subject: [PATCH] Redirect stdout from noisy model conversion --- .../rootfs/etc/s6-overlay/s6-rc.d/trt-model-prepare/run | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/docker/support/tensorrt_detector/rootfs/etc/s6-overlay/s6-rc.d/trt-model-prepare/run b/docker/support/tensorrt_detector/rootfs/etc/s6-overlay/s6-rc.d/trt-model-prepare/run index 602315a93..5f0e43553 100755 --- a/docker/support/tensorrt_detector/rootfs/etc/s6-overlay/s6-rc.d/trt-model-prepare/run +++ b/docker/support/tensorrt_detector/rootfs/etc/s6-overlay/s6-rc.d/trt-model-prepare/run @@ -41,12 +41,13 @@ echo "Generating the following TRT Models: ${MODEL_CONVERT}" cd /usr/local/src/tensorrt_demos/yolo # Download yolo weights -./download_yolo.sh $MODEL_CONVERT +./download_yolo.sh $MODEL_CONVERT > /dev/null for model in ${MODEL_CONVERT//,/ } do - python3 yolo_to_onnx.py -m ${model} - python3 onnx_to_tensorrt.py -m ${model} + echo "Converting ${model} model" + python3 yolo_to_onnx.py -m ${model} > /dev/null + python3 onnx_to_tensorrt.py -m ${model} > /dev/null cp ${model}.trt ${OUTPUT_FOLDER}/${model}.trt ln -s ${OUTPUT_FOLDER}/${model}.trt ${MODEL_CACHE_DIR}/${model}.trt done