Pular para conteúdo

Battery analyser

Battery

Bases: enum.Enum

Classificação das baterias

Attributes:

Name Type Description
V9

Bateria 9V
AA

Pilha AA
AAA

Pilha AAA
C

Pilha C
D

Pilha D
UNKNOWN

Objeto desconhecido
Source code in battery_analyser/predict.py 
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
class Battery(enum.Enum):
    """
    Classificação das baterias

    Attributes:

        V9: Bateria 9V
        AA: Pilha AA
        AAA: Pilha AAA
        C: Pilha C
        D: Pilha D
        UNKNOWN: Objeto desconhecido
    """

    V9 = 0
    AA = 1
    AAA = 2
    C = 3
    D = 4
    UNKNOWN = 5

predict(image)

Classifica uma imagem de uma bateria

Parameters:

Name Type Description Default
image np.ndarray

Array de bytes de 3 dimensões com a imagem

 required

Returns:

Type Description
Battery

Uma Predicton com a classificação da bateria e a porcentagem da predição.
Source code in battery_analyser/predict.py 
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
def predict(image: np.ndarray) -> Battery:
    """
    Classifica uma imagem de uma bateria

    Args:
        image: Array de bytes de 3 dimensões com a imagem

    Returns:
        Uma Predicton com a classificação da bateria e a porcentagem da predição.
    """
    tensor_image = vision.TensorImage.create_from_array(image)
    categories = classifier.classify(tensor_image)
    if not categories.classifications or not categories.classifications[0].categories or categories.classifications[0].categories[0].score < THRESHOLD:
        return Battery.UNKNOWN
    return Battery(categories.classifications[0].categories[0].index)