From 7050112544228113ae650eb9286ce319269b3f61 Mon Sep 17 00:00:00 2001 From: "hongliang.yuan" Date: Fri, 28 Nov 2025 16:44:55 +0800 Subject: [PATCH] fix yolov10,kie_layoutxlm,fcos,wide_and_deep --- .../speech_recognition/transformer_asr/ixrt/requirements.txt | 3 ++- models/cv/object_detection/yolov10/igie/ci/prepare.sh | 2 +- models/cv/ocr/kie_layoutxlm/igie/requirements.txt | 3 ++- models/nlp/plm/videobert/ixrt/README.md | 2 ++ models/others/recommendation/wide_and_deep/ixrt/README.md | 3 +++ models/others/recommendation/wide_and_deep/ixrt/ci/prepare.sh | 3 +++ tests/run_ixrt.py | 2 +- 7 files changed, 14 insertions(+), 4 deletions(-) diff --git a/models/audio/speech_recognition/transformer_asr/ixrt/requirements.txt b/models/audio/speech_recognition/transformer_asr/ixrt/requirements.txt index 4aa5b041..461f731f 100644 --- a/models/audio/speech_recognition/transformer_asr/ixrt/requirements.txt +++ b/models/audio/speech_recognition/transformer_asr/ixrt/requirements.txt @@ -1 +1,2 @@ -speechbrain==0.5.13 \ No newline at end of file +speechbrain==0.5.13 +requests \ No newline at end of file diff --git a/models/cv/object_detection/yolov10/igie/ci/prepare.sh b/models/cv/object_detection/yolov10/igie/ci/prepare.sh index e1e2e0d7..5afd8d9d 100644 --- a/models/cv/object_detection/yolov10/igie/ci/prepare.sh +++ b/models/cv/object_detection/yolov10/igie/ci/prepare.sh @@ -27,7 +27,7 @@ fi pip3 install -r requirements.txt -git clone --depth 1 https://github.com/THU-MIG/yolov10.git +cp -r /mnt/deepspark/data/3rd_party/yolov10 ./ cd yolov10 pip3 install -e . --no-deps diff --git a/models/cv/ocr/kie_layoutxlm/igie/requirements.txt b/models/cv/ocr/kie_layoutxlm/igie/requirements.txt index ae626e20..1055283c 100644 --- a/models/cv/ocr/kie_layoutxlm/igie/requirements.txt +++ b/models/cv/ocr/kie_layoutxlm/igie/requirements.txt @@ -8,4 +8,5 @@ paddleocr==2.6.0 paddle2onnx==1.3.0 python-bidi protobuf==3.20.3 -aistudio-sdk==0.2.6 \ No newline at end of file +aistudio-sdk==0.2.6 +numpy==1.26.4 \ No newline at end of file diff --git a/models/nlp/plm/videobert/ixrt/README.md b/models/nlp/plm/videobert/ixrt/README.md index 01197a1b..69ae046f 100644 --- a/models/nlp/plm/videobert/ixrt/README.md +++ b/models/nlp/plm/videobert/ixrt/README.md @@ -12,6 +12,8 @@ and textual information into a unified framework. | :----: | :----: | :----: | | MR-V100 | 4.2.0 | 25.03 | +**This model is compatible with IXUCA SDK up to version 4.2.0.** + ## Model Preparation ### Prepare Resources diff --git a/models/others/recommendation/wide_and_deep/ixrt/README.md b/models/others/recommendation/wide_and_deep/ixrt/README.md index 19772222..afdf98a5 100644 --- a/models/others/recommendation/wide_and_deep/ixrt/README.md +++ b/models/others/recommendation/wide_and_deep/ixrt/README.md @@ -47,6 +47,9 @@ python3 change2dynamic.py --model_path open_wide_deep_saved_model/widedeep_sim.o ## Model Inference ```bash +git clone https://gitee.com/deep-spark/iluvatar-corex-ixrt.git --depth=1 +cp -r iluvatar-corex-ixrt/tools/optimizer/ ../../../../../toolbox/ByteMLPerf/byte_infer_perf/general_perf/backends/ILUVATAR/ + export ORIGIN_ONNX_NAME=./open_wide_deep_saved_model/widedeep_sim export OPTIMIER_FILE=${IXRT_OSS_ROOT}/tools/optimizer/optimizer.py export PROJ_PATH=./ diff --git a/models/others/recommendation/wide_and_deep/ixrt/ci/prepare.sh b/models/others/recommendation/wide_and_deep/ixrt/ci/prepare.sh index 4a351d87..a200a66f 100644 --- a/models/others/recommendation/wide_and_deep/ixrt/ci/prepare.sh +++ b/models/others/recommendation/wide_and_deep/ixrt/ci/prepare.sh @@ -19,6 +19,9 @@ set -x apt install -y libnuma-dev pip install -r requirements.txt +git clone https://gitee.com/deep-spark/iluvatar-corex-ixrt.git --depth=1 +cp -r iluvatar-corex-ixrt/tools/optimizer/ ../../../../../toolbox/ByteMLPerf/byte_infer_perf/general_perf/backends/ILUVATAR/ + cp -r /root/data/checkpoints/open_wide_deep_saved_model ./ python3 export_onnx.py --model_path open_wide_deep_saved_model --output_path open_wide_deep_saved_model/widedeep.onnx diff --git a/tests/run_ixrt.py b/tests/run_ixrt.py index 2f5c0cc1..542a810d 100644 --- a/tests/run_ixrt.py +++ b/tests/run_ixrt.py @@ -289,7 +289,7 @@ def run_detec_testcase(model): logging.info(f"Start running {model_name} {prec} test case") result["result"].setdefault(prec, {}) result["result"].setdefault(prec, {"status": "FAIL"}) - if model_name in ["yolov3", "yolov5", "yolov5s", "yolov7", "atss", "paa", "retinanet", "yolof"]: + if model_name in ["yolov3", "yolov5", "yolov5s", "yolov7", "atss", "paa", "retinanet", "yolof", "fcos"]: script = f""" cd ../{model['model_path']} export DATASETS_DIR=./{dataset_n}/ -- Gitee