From 7c7537e5df42b9ea84a2c0e39f7cf76bd311901d Mon Sep 17 00:00:00 2001
From: Olegs NIKISINS <onikisins@italix03.idiap.ch>
Date: Wed, 27 Sep 2017 16:21:01 +0200
Subject: [PATCH] Added entry points for VideoHistOfSparseCodes extractor
---
.../extractor/video_hist_of_sparse_codes.py | 16 ++++++++++++++++
bob/pad/face/extractor/__init__.py | 2 ++
setup.py | 6 +++++-
3 files changed, 23 insertions(+), 1 deletion(-)
create mode 100644 bob/pad/face/config/extractor/video_hist_of_sparse_codes.py
diff --git a/bob/pad/face/config/extractor/video_hist_of_sparse_codes.py b/bob/pad/face/config/extractor/video_hist_of_sparse_codes.py
new file mode 100644
index 00000000..6cce7e45
--- /dev/null
+++ b/bob/pad/face/config/extractor/video_hist_of_sparse_codes.py
@@ -0,0 +1,16 @@
+#!/usr/bin/env python
+
+from bob.pad.face.extractor import VideoHistOfSparseCodes
+
+
+#=======================================================================================
+# Define instances here:
+
+METHOD = "mean"
+extractor_mean = VideoHistOfSparseCodes(method = METHOD)
+
+
+METHOD = "hist"
+extractor_hist = VideoHistOfSparseCodes(method = METHOD)
+
+
diff --git a/bob/pad/face/extractor/__init__.py b/bob/pad/face/extractor/__init__.py
index 310989b4..1699d534 100644
--- a/bob/pad/face/extractor/__init__.py
+++ b/bob/pad/face/extractor/__init__.py
@@ -4,6 +4,7 @@ from .ImageQualityMeasure import ImageQualityMeasure
from .VideoDataLoader import VideoDataLoader
from .VideoQualityMeasure import VideoQualityMeasure
from .FrameDiffFeatures import FrameDiffFeatures
+from .VideoHistOfSparseCodes import VideoHistOfSparseCodes
def __appropriate__(*args):
"""Says object was actually declared here, and not in the import module.
@@ -30,5 +31,6 @@ __appropriate__(
VideoQualityMeasure,
VideoDataLoader,
FrameDiffFeatures,
+ VideoHistOfSparseCodes,
)
__all__ = [_ for _ in dir() if not _.startswith('_')]
diff --git a/setup.py b/setup.py
index aea2d186..552001a1 100644
--- a/setup.py
+++ b/setup.py
@@ -108,11 +108,15 @@ setup(
'sparse-coding-preprocessor = bob.pad.face.config.preprocessor.video_sparse_coding:preprocessor',
],
- # registered preprocessors:
+ # registered extractors:
'bob.pad.extractor': [
'video-lbp-histogram-extractor-n8r1-uniform = bob.pad.face.config.extractor.video_lbp_histogram:video_lbp_histogram_extractor_n8r1_uniform',
'video-quality-measure-galbally-msu = bob.pad.face.config.extractor.video_quality_measure:video_quality_measure_galbally_msu',
'frame-diff-feat-extr-w20-over0 = bob.pad.face.config.extractor.frame_diff_features:frame_diff_feat_extr_w20_over0',
+
+ # extractors for sparse coding:
+ 'hist-of-sparse-codes-mean = bob.pad.face.config.extractor.video_hist_of_sparse_codes:extractor_mean',
+ 'hist-of-sparse-codes-hist = bob.pad.face.config.extractor.video_hist_of_sparse_codes:extractor_hist',
],
# registered algorithms:
--
GitLab