Skip to content
Snippets Groups Projects
Select Git revision
  • master default protected
  • beat.backend.python_v1.4.2_1
  • beat.backend.python_v1.4.2
  • v1.0.0
4 results
Blame Permalink
5.py 4.08 KiB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110 






###############################################################################
#                                                                             #
# Copyright (c) 2018 Idiap Research Institute, http://www.idiap.ch/           #
# Contact: beat.support@idiap.ch                                              #
#                                                                             #
# This file is part of the beat.examples module of the BEAT platform.         #
#                                                                             #
# Commercial License Usage                                                    #
# Licensees holding valid commercial BEAT licenses may use this file in       #
# accordance with the terms contained in a written agreement between you      #
# and Idiap. For further information contact tto@idiap.ch                     #
#                                                                             #
# Alternatively, this file may be used under the terms of the GNU Affero      #
# Public License version 3 as published by the Free Software and appearing    #
# in the file LICENSE.AGPL included in the packaging of this file.            #
# The BEAT platform is distributed in the hope that it will be useful, but    #
# WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY  #
# or FITNESS FOR A PARTICULAR PURPOSE.                                        #
#                                                                             #
# You should have received a copy of the GNU Affero Public License along      #
# with the BEAT platform. If not, see http://www.gnu.org/licenses/.           #
#                                                                             #
###############################################################################

import os
import numpy as np
from collections import namedtuple

from beat.backend.python.database import View

import bob.io.base
import bob.io.image
from bob.db.livdet2013 import Database


#----------------------------------------------------------


class All(View):
    """Outputs:
        - image: "{{ system_user.username }}/array_3d_uint8/1"
        - spoof: "{{ system_user.username }}/boolean/1"

    Several "image" are associated with a given "spoof".

    --------------- --------------- --------------- --------------- 
    |    image    | |    image    | |    image    | |    image    | 
    --------------- --------------- --------------- --------------- 
    ------------------------------- ------------------------------ 
    |            spoof            | |            spoof            |
    ------------------------------- ------------------------------ 
    """

    def index(self, root_folder, parameters):
        Entry = namedtuple('Entry', ['spoof', 'image'])

        # Open the database and load the objects to provide via the outputs
        db = Database()
        objs = sorted(db.objects(protocols=parameters.get('protocol'),
                                 groups=parameters['group'],
                                 classes=parameters.get('class')),
                      key=lambda x: x.is_live())

        return [ Entry(x.is_live(), x.make_path(root_folder)) for x in objs ]


    def get(self, output, index):
        obj = self.objs[index]

        if output == 'spoof':
            return {
                'value': obj.spoof
            }

        elif output == 'image':
            return {
                'value': bob.io.base.load(obj.image)
            }


#----------------------------------------------------------


def setup_tests():
    # Install a mock load function for the images
    def mock_load(root_folder):
        return np.ndarray((3, 10, 20), dtype=np.uint8)

    bob.io.base.load = mock_load


#----------------------------------------------------------


# Test the behavior of the views (on fake data)
if __name__ == '__main__':

    setup_tests()

    view = All()
    view.objs = view.index(
        root_folder='',
        parameters=dict(
            protocol='Biometrika',
            group='train'
        )
    )
    view.get('spoof', 0)
    view.get('image', 0)