TensorRT  7.2.1.6
NVIDIA TensorRT
Looking for a C++ dev who knows TensorRT?
I'm looking for work. Hire me!
All Classes Namespaces Functions Variables Typedefs Enumerations Enumerator Friends Pages
mrcnn_to_trt_single Namespace Reference

Classes

class  CocoConfig
 
class  InferenceConfig
 

Functions

def parse_command_line_arguments (args=None)
 
def main (args=None)
 
def convert_model (inference_model, output_path, output_nodes=[], preprocessor=None, text=False, list_nodes=False)
 

Variables

 ROOT_DIR = os.path.abspath("./")
 
 LOG_DIR = os.path.join(ROOT_DIR, "logs")
 

Function Documentation

◆ parse_command_line_arguments()

def mrcnn_to_trt_single.parse_command_line_arguments (   args = None)
Here is the caller graph for this function:

◆ main()

def mrcnn_to_trt_single.main (   args = None)
Here is the call graph for this function:

◆ convert_model()

def mrcnn_to_trt_single.convert_model (   inference_model,
  output_path,
  output_nodes = [],
  preprocessor = None,
  text = False,
  list_nodes = False 
)
Here is the caller graph for this function:

Variable Documentation

◆ ROOT_DIR

mrcnn_to_trt_single.ROOT_DIR = os.path.abspath("./")

◆ LOG_DIR

mrcnn_to_trt_single.LOG_DIR = os.path.join(ROOT_DIR, "logs")