Apollo  6.0
Open source self driving car software
apollo::perception::inference::RTNet Member List

This is the complete list of members for apollo::perception::inference::RTNet, including all inherited members.

addActiveLayer(const LayerParameter &layer_param, nvinfer1::ITensor *const *inputs, int nbInputs, nvinfer1::INetworkDefinition *net, TensorMap *tensor_map, TensorModifyMap *tensor_modify_map)apollo::perception::inference::RTNetprotected
addArgmaxLayer(const LayerParameter &layer_param, nvinfer1::ITensor *const *inputs, int nbInputs, nvinfer1::INetworkDefinition *net, TensorMap *tensor_map, TensorModifyMap *tensor_modify_map)apollo::perception::inference::RTNetprotected
addBatchnormLayer(const LayerParameter &layer_param, nvinfer1::ITensor *const *inputs, WeightMap *weight_map, nvinfer1::INetworkDefinition *net, TensorMap *tensor_map, TensorModifyMap *tensor_modify_map)apollo::perception::inference::RTNetprotected
addConcatLayer(const LayerParameter &layer_param, nvinfer1::ITensor *const *inputs, int nbInputs, nvinfer1::INetworkDefinition *net, TensorMap *tensor_map, TensorModifyMap *tensor_modify_map)apollo::perception::inference::RTNetprotected
addConvLayer(const LayerParameter &layer_param, nvinfer1::ITensor *const *inputs, WeightMap *weight_map, nvinfer1::INetworkDefinition *net, TensorMap *tensor_map, TensorModifyMap *tensor_modify_map)apollo::perception::inference::RTNetprotected
addDeconvLayer(const LayerParameter &layer_param, nvinfer1::ITensor *const *inputs, WeightMap *weight_map, nvinfer1::INetworkDefinition *net, TensorMap *tensor_map, TensorModifyMap *tensor_modify_map)apollo::perception::inference::RTNetprotected
addDFMBPSROIAlignLayer(const LayerParameter &layer_param, nvinfer1::ITensor *const *inputs, int nbInputs, nvinfer1::INetworkDefinition *net, TensorMap *tensor_map, TensorModifyMap *tensor_modify_map)apollo::perception::inference::RTNetprotected
addEltwiseLayer(const LayerParameter &layer_param, nvinfer1::ITensor *const *inputs, nvinfer1::INetworkDefinition *net, TensorMap *tensor_map, TensorModifyMap *tensor_modify_map)apollo::perception::inference::RTNetprotected
addInnerproductLayer(const LayerParameter &layer_param, nvinfer1::ITensor *const *inputs, WeightMap *weight_map, nvinfer1::INetworkDefinition *net, TensorMap *tensor_map, TensorModifyMap *tensor_modify_map)apollo::perception::inference::RTNetprotected
addInput(const TensorDimsMap &tensor_dims_map, const std::map< std::string, std::vector< int >> &shapes, TensorMap *tensor_map)apollo::perception::inference::RTNetprotected
addLayer(const LayerParameter &layer_param, nvinfer1::ITensor *const *inputs, int nbInputs, WeightMap *weight_map, nvinfer1::INetworkDefinition *net, TensorMap *tensor_map, TensorModifyMap *tensor_modify_map)apollo::perception::inference::RTNetprotected
addPaddingLayer(const LayerParameter &layer_param, nvinfer1::ITensor *const *inputs, nvinfer1::INetworkDefinition *net, TensorMap *tensor_map, TensorModifyMap *tensor_modify_map)apollo::perception::inference::RTNetprotected
addPermuteLayer(const LayerParameter &layer_param, nvinfer1::ITensor *const *inputs, int nbInputs, nvinfer1::INetworkDefinition *net, TensorMap *tensor_map, TensorModifyMap *tensor_modify_map)apollo::perception::inference::RTNetprotected
addPoolingLayer(const LayerParameter &layer_param, nvinfer1::ITensor *const *inputs, nvinfer1::INetworkDefinition *net, TensorMap *tensor_map, TensorModifyMap *tensor_modify_map)apollo::perception::inference::RTNetprotected
addRCNNProposalLayer(const LayerParameter &layer_param, nvinfer1::ITensor *const *inputs, int nbInputs, nvinfer1::INetworkDefinition *net, TensorMap *tensor_map, TensorModifyMap *tensor_modify_map)apollo::perception::inference::RTNetprotected
addReshapeLayer(const LayerParameter &layer_param, nvinfer1::ITensor *const *inputs, nvinfer1::INetworkDefinition *net, TensorMap *tensor_map, TensorModifyMap *tensor_modify_map)apollo::perception::inference::RTNetprotected
addRPNProposalSSDLayer(const LayerParameter &layer_param, nvinfer1::ITensor *const *inputs, int nbInputs, nvinfer1::INetworkDefinition *net, TensorMap *tensor_map, TensorModifyMap *tensor_modify_map)apollo::perception::inference::RTNetprotected
addScaleLayer(const LayerParameter &layer_param, nvinfer1::ITensor *const *inputs, WeightMap *weight_map, nvinfer1::INetworkDefinition *net, TensorMap *tensor_map, TensorModifyMap *tensor_modify_map)apollo::perception::inference::RTNetprotected
addSliceLayer(const LayerParameter &layer_param, nvinfer1::ITensor *const *inputs, int nbInputs, nvinfer1::INetworkDefinition *net, TensorMap *tensor_map, TensorModifyMap *tensor_modify_map)apollo::perception::inference::RTNetprotected
addSoftmaxLayer(const LayerParameter &layer_param, nvinfer1::ITensor *const *inputs, int nbInputs, nvinfer1::INetworkDefinition *net, TensorMap *tensor_map, TensorModifyMap *tensor_modify_map)apollo::perception::inference::RTNetprotected
checkInt8(const std::string &gpu_name, nvinfer1::IInt8Calibrator *calibrator)apollo::perception::inference::RTNetprotected
ConstructMap(const LayerParameter &layer_param, nvinfer1::ILayer *layer, TensorMap *tensor_map, TensorModifyMap *tensor_modify_map)apollo::perception::inference::RTNetprotected
get_blob(const std::string &name) overrideapollo::perception::inference::RTNetvirtual
gpu_id_apollo::perception::inference::Inferenceprotected
Infer() overrideapollo::perception::inference::RTNetvirtual
Inference()=defaultapollo::perception::inference::Inference
Init(const std::map< std::string, std::vector< int >> &shapes) overrideapollo::perception::inference::RTNetvirtual
init_blob(std::vector< std::string > *names)apollo::perception::inference::RTNetprotected
loadLayerWeights(const float *data, int size)apollo::perception::inference::RTNetprotected
loadLayerWeights(float data, int size)apollo::perception::inference::RTNetprotected
loadWeights(const std::string &model_file, WeightMap *weight_map)apollo::perception::inference::RTNetprotected
max_batch_size_apollo::perception::inference::Inferenceprotected
mergeBN(int index, LayerParameter *layer_param)apollo::perception::inference::RTNetprotected
parse_with_api(const std::map< std::string, std::vector< int >> &shapes)apollo::perception::inference::RTNetprotected
RTNet(const std::string &net_file, const std::string &model_file, const std::vector< std::string > &outputs, const std::vector< std::string > &inputs)apollo::perception::inference::RTNet
RTNet(const std::string &net_file, const std::string &model_file, const std::vector< std::string > &outputs, const std::vector< std::string > &inputs, nvinfer1::Int8EntropyCalibrator *calibrator)apollo::perception::inference::RTNet
RTNet(const std::string &net_file, const std::string &model_file, const std::vector< std::string > &outputs, const std::vector< std::string > &inputs, const std::string &model_root)apollo::perception::inference::RTNet
set_gpu_id(const int &gpu_id)apollo::perception::inference::Inference
set_max_batch_size(const int &batch_size)apollo::perception::inference::Inference
shape(const std::string &name, std::vector< int > *res)apollo::perception::inference::RTNetprotected
~Inference()=defaultapollo::perception::inference::Inferencevirtual
~RTNet()apollo::perception::inference::RTNetvirtual