update predict for v9\v10

mindspore-lab · Feb 20, 2025 · eeb2ec5 · eeb2ec5
1 parent bdc48d6
commit eeb2ec5
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
@@ -17,7 +17,7 @@ jobs:
     strategy:
       matrix:
         os: [ubuntu-latest] # , macos-latest, windows-latest
-        python: [3.7, 3.8, 3.9]
+        python: [3.8, 3.9]
     runs-on: ${{ matrix.os }}
     steps:
     - uses: actions/checkout@v3

diff --git a/demo/predict.py b/demo/predict.py
@@ -124,7 +124,7 @@ def detect(
 
     # Run infer
     _t = time.time()
-    out = network(imgs_tensor)  # inference and training outputs
+    out, _ = network(imgs_tensor)  # inference and training outputs
     out = out[0] if isinstance(out, (tuple, list)) else out
     infer_times = time.time() - _t