Skip to content

pc

Classes

PcBackend 

PcBackend(params: HKTaskParams)

PC inference engine backend.

This backend runs inference on a PC using Keras or TFLite models.

Parameters:

Source code in heartkit/backends/pc.py 
10
11
12
13
14
15
16
17
18
19
20
21
22
def __init__(self, params: HKTaskParams) -> None:
    """PC inference engine backend.

    This backend runs inference on a PC using Keras or TFLite models.

    Args:
        params (HKTaskParams): Task parameters
    """

    super().__init__(params=params)
    self._inputs = None
    self._outputs = None
    self._model = None

Functions

open 
open()

This method will simply load the keras or TFLite model

Source code in heartkit/backends/pc.py 
28
29
30
31
32
33
34
35
def open(self):
    """This method will simply load the keras or TFLite model"""
    if self._is_tf_model():
        self._model = helia.models.load_model(self.params.model_file)
    else:
        with open(self.params.model_file, "rb") as fp:
            model_content = fp.read()
        self._model = helia.interpreters.tflite.TfLiteKerasInterpreter(model_content=model_content)
close 
close()

This method will unload the model

Source code in heartkit/backends/pc.py 
37
38
39
def close(self):
    """This method will unload the model"""
    self._model = None
set_inputs 
set_inputs(inputs: NDArray)

Set inputs for inference

Parameters:

  • inputs
    (NDArray) –

    Inputs for inference

Source code in heartkit/backends/pc.py 
41
42
43
44
45
46
47
def set_inputs(self, inputs: npt.NDArray):
    """Set inputs for inference

    Args:
        inputs (npt.NDArray): Inputs for inference
    """
    self._inputs = inputs
perform_inference 
perform_inference()

Perform inference on the model

Source code in heartkit/backends/pc.py 
49
50
51
52
53
54
def perform_inference(self):
    """Perform inference on the model"""
    if self._is_tf_model():
        self._outputs = self._model.predict(np.expand_dims(self._inputs, 0), verbose=0).squeeze(0)
    else:
        self._outputs = self._model.predict(self._inputs)
get_outputs 
get_outputs() -> npt.NDArray

Get outputs from inference

Returns:

  • NDArray –

    npt.NDArray: Outputs from inference

Source code in heartkit/backends/pc.py 
56
57
58
59
60
61
62
def get_outputs(self) -> npt.NDArray:
    """Get outputs from inference

    Returns:
        npt.NDArray: Outputs from inference
    """
    return self._outputs