MCPcopy Create free account

hub / github.com/NVIDIA-AI-IOT/tf_trt_models / functions

Functions19 in github.com/NVIDIA-AI-IOT/tf_trt_models

⨍ Functions 19◇ Types & classes 0
↓ 2 callersFunctionconvert_relu6
(graph_def, const6_name='const6')
tf_trt_models/graph_utils.py:41
↓ 1 callersFunctionmake_const6
(const6_name='const6')
tf_trt_models/graph_utils.py:4
↓ 1 callersFunctionmake_relu6
(output_name, input_name, const6_name='const6')
tf_trt_models/graph_utils.py:11
↓ 1 callersFunctionremove_node
(graph_def, node)
tf_trt_models/graph_utils.py:62
↓ 1 callersFunctionremove_op
(graph_def, op_name)
tf_trt_models/graph_utils.py:72
Function_mobilenet_v1_0p25_128
(*args, **kwargs)
tf_trt_models/classification.py:34
Function_mobilenet_v1_0p5_160
(*args, **kwargs)
tf_trt_models/classification.py:29
Function_mobilenet_v1_1p0_224
(*args, **kwargs)
tf_trt_models/classification.py:24
Function_preprocess_inception
(x)
tf_trt_models/classification.py:45
Function_preprocess_vgg
(x)
tf_trt_models/classification.py:39
Functionbuild_classification_graph
Builds an image classification model by name This function builds an image classification model given a model name, parameter checkpoint file
tf_trt_models/classification.py:169
Functionbuild_detection_graph
Builds a frozen graph for a pre-trained object detection model
tf_trt_models/detection.py:103
Functiondownload_classification_checkpoint
Downloads an image classification model pretrained checkpoint by name :param model: the model name (see table) :type model: string :param
tf_trt_models/classification.py:138
Functiondownload_detection_model
Downloads a pre-trained object detection model
tf_trt_models/detection.py:80
Functionforce_nms_cpu
(frozen_graph)
tf_trt_models/graph_utils.py:78
Functionget_input_names
(model)
tf_trt_models/detection.py:69
Functionget_output_names
(model)
tf_trt_models/detection.py:73
Functionremove_assert
(frozen_graph)
tf_trt_models/graph_utils.py:89
Functionreplace_relu6
(frozen_graph)
tf_trt_models/graph_utils.py:85