Skip to content

Commit 23c329e

Browse files
gaze_estimation_demo landmarks-98 (#3327)
* enable facials-landmarks-98 in models.lst * add test case for facial-landmarks-98
1 parent 4a71e2e commit 23c329e

File tree

2 files changed

+28
-20
lines changed

2 files changed

+28
-20
lines changed

demos/gaze_estimation_demo/cpp/models.lst

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
# This file can be used with the --list option of the model downloader.
2-
#facial-landmarks-98-detection-????
32
facial-landmarks-35-adas-????
3+
facial-landmarks-98-detection-????
44
face-detection-adas-????
55
face-detection-retail-????
66
gaze-estimation-adas-????

demos/tests/cases.py

Lines changed: 27 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -318,25 +318,33 @@ def single_option_cases(key, *args):
318318
),
319319
)),
320320

321-
# TODO: FP16-INT8 expected to have output named "angle_y_fc"
322-
# CppDemo(name='gaze_estimation_demo',
323-
# model_keys=['-m', '-m_fd', '-m_hp', '-m_lm', '-m_es'],
324-
# device_keys=['-d', '-d_fd', '-d_hp', '-d_lm'],
325-
# test_cases=combine_cases(
326-
# TestCase(options={'-no_show': None,
327-
# **MONITORS,
328-
# '-i': DataPatternArg('gaze-estimation-adas')}),
329-
# TestCase(options={
330-
# '-m': ModelArg('gaze-estimation-adas-0002'),
331-
# '-m_hp': ModelArg('head-pose-estimation-adas-0001'),
332-
# '-m_lm': ModelArg('facial-landmarks-35-adas-0002'),
333-
# '-m_es': ModelArg('open-closed-eye-0001'),
334-
# }),
335-
# single_option_cases(
336-
# '-m_fd',
337-
# ModelArg('face-detection-adas-0001'),
338-
# ModelArg('face-detection-retail-0004')),
339-
# )),
321+
CppDemo(name='gaze_estimation_demo',
322+
model_keys=['-m', '-m_fd', '-m_hp', '-m_lm', '-m_es'],
323+
device_keys=['-d', '-d_fd', '-d_hp', '-d_lm'],
324+
test_cases=combine_cases(
325+
TestCase(options={'-no_show': None,
326+
**MONITORS,
327+
'-i': DataPatternArg('gaze-estimation-adas')}),
328+
TestCase(options={
329+
'-m': ModelArg('gaze-estimation-adas-0002'),
330+
'-m_hp': ModelArg('head-pose-estimation-adas-0001'),
331+
'-m_es': ModelArg('open-closed-eye-0001'),
332+
}),
333+
[
334+
*combine_cases(
335+
single_option_cases('-m_lm',
336+
ModelArg('facial-landmarks-35-adas-0002'),
337+
ModelArg('facial-landmarks-98-detection-0001'),
338+
)),
339+
],
340+
[
341+
*combine_cases(
342+
single_option_cases('-m_fd',
343+
ModelArg('face-detection-adas-0001'),
344+
ModelArg('face-detection-retail-0004')
345+
)),
346+
],
347+
)),
340348

341349
CppDemo(name='human_pose_estimation_demo', device_keys=['-d'], test_cases=combine_cases(
342350
TestCase(options={'-no_show': None,

0 commit comments

Comments
 (0)