Skip to content

Commit

Permalink
Tagging ultralytics image
Browse files Browse the repository at this point in the history
Signed-off-by: Antonio Martinez <jose.antonio.martinez.torres@intel.com>
  • Loading branch information
antoniomtz committed Oct 25, 2024
1 parent 17df0b5 commit 94fc7d3
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 2 deletions.
2 changes: 1 addition & 1 deletion Makefile
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@ download-yolov8s:
docker run --user 1000:1000 -e HTTPS_PROXY=${HTTPS_PROXY} -e HTTP_PROXY=${HTTPS_PROXY} --rm \
-e YOLO_DEBUG=1 \
-v $(PWD)/models/object_detection/yolov8s:/models \
ultralytics/ultralytics:latest-cpu \
ultralytics/ultralytics:8.2.101-cpu \
bash -c "cd /models && yolo export model=yolov8s.pt format=openvino"; \
mv $(PWD)/models/object_detection/yolov8s/yolov8s_openvino_model $(PWD)/models/object_detection/yolov8s/FP32; \
else \
Expand Down
2 changes: 1 addition & 1 deletion src/pipelines/yolov8s_roi.sh
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@ fi
echo "decode type $DECODE"
echo "Run YOLOv8 pipeline with ROI and Filtering on $DEVICE with batch size = $BATCH_SIZE"

gstLaunchCmd="GST_DEBUG=\"GST_TRACER:7\" GST_TRACERS=\"latency_tracer(flags=pipeline,interval=100)\" gst-launch-1.0 $inputsrc ! $DECODE ! gvaattachroi mode=1 file-path=/home/pipeline-server/pipelines/roi.json ! gvadetect batch-size=$BATCH_SIZE model-instance-id=odmodel name=detection model=models/object_detection/yolov8s/FP32/yolov8s.xml device=$DEVICE $PRE_PROCESS inference-region=1 object-class=$ROI threshold=0.5 ! \
gstLaunchCmd="GST_DEBUG=1 GST_TRACERS=\"latency_tracer(flags=pipeline,interval=100)\" gst-launch-1.0 $inputsrc ! $DECODE ! gvaattachroi mode=1 file-path=/home/pipeline-server/pipelines/roi.json ! gvadetect batch-size=$BATCH_SIZE model-instance-id=odmodel name=detection model=models/object_detection/yolov8s/FP32/yolov8s.xml device=$DEVICE $PRE_PROCESS inference-region=1 object-class=$ROI threshold=0.5 ! \
gvapython module=/home/pipeline-server/extensions/object_filter.py class=ObjectDetectionFilter kwarg=\"{\\\"class_ids\\\": \\\"$CLASS_IDS\\\", \\\"rois\\\": \\\"$ROI\\\"}\" ! gvatrack ! \
$AGGREGATE gvametaconvert name=metaconvert add-empty-results=true ! \
gvapython module=/home/pipeline-server/extensions/gva_roi_metadata.py class=RoiMetadata kwarg=\"{\\\"roi\\\": \\\"$ROI\\\"}\" ! \
Expand Down

0 comments on commit 94fc7d3

Please sign in to comment.