Skip to content

model_handler

Modelhandler

Bases: ModelHandlerInterface

Handle

Parameters:

Name Type Description Default
ModelHandlerInterface [type]

[description]

 required

Raises:

Type Description
NoImageModelSpesifiedError

[description]

Returns:

Type Description
Source code in src/artificial_artwork/pretrained_model/model_handler.py 
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
class Modelhandler(ModelHandlerInterface):
    """Handle

    Args:
        ModelHandlerInterface ([type]): [description]

    Raises:
        NoImageModelSpesifiedError: [description]

    Returns:
        [type]: [description]
    """

    _reporter: ReporterProtocol

    def __init__(self):
        self._reporter = None

    @property
    def reporter(self) -> ReporterProtocol:
        """Reporter getter, which can return Layer Weights given a layer (str) ID

        Args:
            layers ([type]): [description]
        """
        return self._reporter

    @reporter.setter
    def reporter(self, layers) -> None:
        self._reporter = self._create_reporter(layers)

    def _create_reporter(self, layers: NDArray) -> ReporterProtocol:
        return ModelReporter(
            self.model_routines.get_layers_dict(layers), self.model_routines.get_weights
        )

    def load_model_layers(self) -> NDArray:
        layers = self._load_model_layers()
        self._reporter = self._create_reporter(layers)
        return layers

    def _load_model_layers(self) -> NDArray:
        try:
            return self.model_routines.load_layers(os.environ[self.environment_variable])
        except KeyError as variable_not_found:
            raise NoImageModelSpesifiedError(
                self.model_load_exception_text
            ) from variable_not_found

reporter: ReporterProtocol property writable

Reporter getter, which can return Layer Weights given a layer (str) ID

Parameters:

Name Type Description Default
layers [type]

[description]

 required