Skip to content

Commit

Permalink
fix: address PR comments
Browse files Browse the repository at this point in the history
Signed-off-by: Jim Wang <[email protected]>
  • Loading branch information
jim-wang-intel committed Feb 16, 2024
1 parent 08158fe commit 05ead21
Show file tree
Hide file tree
Showing 6 changed files with 22 additions and 4 deletions.
2 changes: 1 addition & 1 deletion configs/opencv-ovms/envs/capi_yolov8_ensemble.env
Original file line number Diff line number Diff line change
Expand Up @@ -11,5 +11,5 @@ RENDER_MODE=1
cl_cache_dir=/home/intel/gst-ovms/.cl-cache
WINDOW_WIDTH=1280
WINDOW_HEIGHT=720
DETECTION_THRESHOLD=0.7
DETECTION_THRESHOLD=0.5
DC=0
1 change: 1 addition & 0 deletions docs_src/OVMS/capiYolov8EnsemblePipelineRun.md
Original file line number Diff line number Diff line change
Expand Up @@ -41,6 +41,7 @@ You can add multiple environment variable files to `configs/opencv-ovms/envs/` d
| WINDOW_HEIGHT | 720 | display window height |
| DETECTION_THRESHOLD | 0.7 | detection threshold value in floating point that needs to be between 0.0 to 1.0 |
| INPUT_TYPE | RTSP_H264 | video input type to us, value: RTSP_H264, FILE_H264, RTSP_H265, or FILE_H265 |
| DC | 0 | direct console output results, value: 0 or 1 |


details of yolov8s pipeline environment variable file can be viewed in [`configs/opencv-ovms/envs/capi_yolov8_ensemble.env`](https://github.com/intel-retail/automated-self-checkout/blob/main/configs/opencv-ovms/envs/capi_yolov8_ensemble.env).
Expand Down
6 changes: 5 additions & 1 deletion download_models/model_build/convert-model.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,8 @@

#
# Copyright (C) 2024 Intel Corporation.
#
# SPDX-License-Identifier: Apache-2.0
#

from pathlib import Path
from typing import Tuple, Dict
Expand Down
5 changes: 5 additions & 0 deletions download_models/model_build/convert-model.sh
Original file line number Diff line number Diff line change
@@ -1,4 +1,9 @@
#!/bin/bash
#
# Copyright (C) 2024 Intel Corporation.
#
# SPDX-License-Identifier: Apache-2.0
#

if [ "$1" == "--refresh" ]
then
Expand Down
5 changes: 5 additions & 0 deletions download_models/model_build/download-tools.sh
Original file line number Diff line number Diff line change
@@ -1,4 +1,9 @@
#!/bin/bash
#
# Copyright (C) 2024 Intel Corporation.
#
# SPDX-License-Identifier: Apache-2.0
#

# tqdm needed?
pip install -q "torch>=2.1" "torchvision>=0.16" "ultralytics==8.0.43" onnx --extra-index-url https://download.pytorch.org/whl/cpu
Expand Down
7 changes: 5 additions & 2 deletions download_models/model_build/quantize-model.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,8 @@


#
# Copyright (C) 2024 Intel Corporation.
#
# SPDX-License-Identifier: Apache-2.0
#

import os
import threading
Expand Down

0 comments on commit 05ead21

Please sign in to comment.