Add python sample for face detection and age classification using HF models#595
Merged
Add python sample for face detection and age classification using HF models#595
Conversation
![github-actions[bot]](https://avatars.githubusercontent.com/in/15368?s=60&v=4){kind=small}
samples/gstreamer/python/face_detection_and_classification/face_detection_and_classification.py
Outdated
Show resolved
Hide resolved
samples/gstreamer/python/face_detection_and_classification/face_detection_and_classification.py
Outdated
Show resolved
Hide resolved
samples/gstreamer/python/face_detection_and_classification/face_detection_and_classification.py
Outdated
Show resolved
Hide resolved
samples/gstreamer/python/face_detection_and_classification/face_detection_and_classification.py
Outdated
Show resolved
Hide resolved
samples/gstreamer/python/face_detection_and_classification/face_detection_and_classification.py
Outdated
Show resolved
Hide resolved
| gi.require_version("GstAnalytics", "1.0") | ||
| from gi.repository import GLib, Gst, GstAnalytics | ||
| from huggingface_hub import hf_hub_download | ||
| from ultralytics import YOLO |
There was a problem hiding this comment.
[pylint] reported by reviewdog 🐶
C0413: Import "from ultralytics import YOLO" should be placed at the top of the module (wrong-import-position)
| from ultralytics import YOLO | ||
|
|
||
| # wrapper to run the gstreamer pipeline loop | ||
| def pipeline_loop(pipeline): |
There was a problem hiding this comment.
[pylint] reported by reviewdog 🐶
C0116: Missing function or method docstring (missing-function-docstring)
samples/gstreamer/python/face_detection_and_classification/face_detection_and_classification.py
Outdated
Show resolved
Hide resolved
samples/gstreamer/python/face_detection_and_classification/face_detection_and_classification.py
Outdated
Show resolved
Hide resolved
samples/gstreamer/python/face_detection_and_classification/face_detection_and_classification.py
Outdated
Show resolved
Hide resolved
| sys.stderr.write("Input video file does not exist\n") | ||
| sys.exit(1) | ||
| else: | ||
| default_video_url = "https://videos.pexels.com/video-files/18553046/18553046-hd_1280_720_30fps.mp4" |
There was a problem hiding this comment.
[pylint] reported by reviewdog 🐶
C0301: Line too long (107/100) (line-too-long)
samples/gstreamer/python/face_detection_and_classification/face_detection_and_classification.py
Outdated
Show resolved
Hide resolved
samples/gstreamer/python/face_detection_and_classification/face_detection_and_classification.py
Outdated
Show resolved
Hide resolved
samples/gstreamer/python/face_detection_and_classification/face_detection_and_classification.py
Outdated
Show resolved
Hide resolved
samples/gstreamer/python/face_detection_and_classification/face_detection_and_classification.py
Outdated
Show resolved
Hide resolved
samples/gstreamer/python/face_detection_and_classification/face_detection_and_classification.py
Outdated
Show resolved
Hide resolved
samples/gstreamer/python/face_detection_and_classification/face_detection_and_classification.py
Outdated
Show resolved
Hide resolved
samples/gstreamer/python/face_detection_and_classification/face_detection_and_classification.py
Outdated
Show resolved
Hide resolved
samples/gstreamer/python/face_detection_and_classification/face_detection_and_classification.py
Outdated
Show resolved
Hide resolved
samples/gstreamer/python/face_detection_and_classification/face_detection_and_classification.py
Outdated
Show resolved
Hide resolved
| import subprocess | ||
| import urllib.request | ||
| import gi | ||
| from gi.repository import Gst |
There was a problem hiding this comment.
[pylint] reported by reviewdog 🐶
E0611: No name 'Gst' in module 'gi.repository' (no-name-in-module)
|
|
||
|
|
||
| # Prepare input video file; download default if none provided | ||
| def prepare_input_video(args): |
There was a problem hiding this comment.
[pylint] reported by reviewdog 🐶
C0116: Missing function or method docstring (missing-function-docstring)
|
|
||
|
|
||
| # wrapper to run the gstreamer pipeline loop | ||
| def pipeline_loop(pipeline): |
There was a problem hiding this comment.
[pylint] reported by reviewdog 🐶
C0116: Missing function or method docstring (missing-function-docstring)
|
|
||
|
|
||
| # Download PyTorch models, convert to OpenVINO IR, create and run gstreamer pipeline | ||
| def main(input_video): |
There was a problem hiding this comment.
[pylint] reported by reviewdog 🐶
C0116: Missing function or method docstring (missing-function-docstring)
tjanczak
commented
Feb 9, 2026
samples/gstreamer/python/face_detection_and_classification/README.md
Outdated
Show resolved
Hide resolved
samples/gstreamer/python/face_detection_and_classification/README.md
Outdated
Show resolved
Hide resolved
msmiatac
requested review from
BaoHuiling,
OskarFiedot,
ZiningLi,
dmichalo,
jmotow,
marcin-wadolkowski,
mholowni,
msmiatac,
nszczygl9,
oonyshch,
pbartosik,
qianlongding,
tbujewsk,
walidbarakat,
yangjianfeng1208 and
yunowo
as code owners
dmichalo
approved these changes
Feb 9, 2026
jmotow
approved these changes
Feb 10, 2026
tbujewsk
reviewed
Feb 10, 2026
| Use optimum-cli to download the face age classifier from Hugging Face and export it to OpenVINO IR. | ||
|
|
||
| **STEP 3 — Build and run the pipeline** | ||
| Construct a GStreamer pipeline with `gvadetect` and `gvaclassify`, run inference, and write an annotated MP4 output. |
Contributor
There was a problem hiding this comment.
add mention about DL Streamer -->
'Construct a GStreamer pipeline with DL Streamer gvadetect and gvaclassify elements'...
tbujewsk
reviewed
Feb 10, 2026
|
|
||
| ## Sample Output | ||
|
|
||
| The script prints the pipeline string and produces an output video annotated with detections and classification results. |
Contributor
There was a problem hiding this comment.
Consider adding screenshot from the sample, to visualize the final effect.
tbujewsk
approved these changes
Feb 10, 2026
Contributor
tbujewsk
left a comment
tbujewsk
left a comment
There was a problem hiding this comment.
Looks good but please see my comments.
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
5 participants
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
…models.
Description
Please include a summary of the changes and the related issue. List any dependencies that are required for this change.
Fixes # (issue)
Any Newly Introduced Dependencies
Please describe any newly introduced 3rd party dependencies in this change. List their name, license information and how they are used in the project.
How Has This Been Tested?
Please describe the tests that you ran to verify your changes. Provide instructions so we can reproduce. Please also list any relevant details for your test configuration
Checklist: