Python shape_tuple Examples

Example #1

def get_faces(result):
    assert len(result.tensors) == 1
    tensor = result.tensors['embeddings']
    print(utils.shape_tuple(tensor.shape))
    # assert utils.shape_tuple(tensor.shape) == (1, 1, 1, 2024)
    return tuple(tensor.data)

Example #2

def _get_probs(result):
    assert len(result.tensors) == 1
    tensor = result.tensors[_OUTPUT_TENSOR_NAME_MAP[result.model_name]]
    assert utils.shape_tuple(tensor.shape) == (1, 1, 1, len(CLASSES))
    return tuple(tensor.data)

Example #3

File: cifar10_classification.py Project: GianMarcoZampa/ProgettoVisionKit

def _get_probs(result):
    assert len(result.tensors) == 1
    tensor = result.tensors['result/Softmax']
    assert utils.shape_tuple(tensor.shape) == (1, 1, 1, len(_CLASSES))
    return tuple(tensor.data)

Example #4

File: dish_classification.py Project: akshayvaidya61/Tensorflow_MobileNet

def _get_probs(result):
    assert len(result.tensors) == 1
    tensor = result.tensors['MobilenetV1/Predictions/Softmax']
    assert utils.shape_tuple(tensor.shape) == (1, 1, 1, 2024)
    return tuple(tensor.data)

Example #5

File: dish_classification.py Project: eosurman/physicsc

def _get_probs(result):
    assert len(result.tensors) == 1
    tensor = result.tensors['MobilenetV1/Predictions/Softmax']
    assert utils.shape_tuple(tensor.shape) == (1, 1, 1, 2024)
    return tuple(tensor.data)