Apollo
6.0
Open source self driving car software
|
#include <map>
#include <string>
#include <vector>
#include "NvCaffeParser.h"
#include "NvInfer.h"
#include <cudnn.h>
#include "modules/perception/proto/rt.pb.h"
#include "cyber/common/log.h"
#include "modules/perception/base/common.h"
Go to the source code of this file.
Classes | |
struct | apollo::perception::inference::ConvParam |
Namespaces | |
apollo | |
PlanningContext is the runtime context in planning. It is persistent across multiple frames. | |
apollo::perception | |
apollo::perception | |
apollo::perception::inference | |
Typedefs | |
typedef std::map< std::string, std::vector< nvinfer1::Weights > > | apollo::perception::inference::WeightMap |
typedef std::map< std::string, nvinfer1::ITensor * > | apollo::perception::inference::TensorMap |
typedef std::map< std::string, nvinfer1::DimsCHW > | apollo::perception::inference::TensorDimsMap |
Functions | |
nvinfer1::DimsCHW | apollo::perception::inference::ReshapeDims (const nvinfer1::DimsCHW &dims, const nvinfer1::DimsCHW &inputDims) |
void | apollo::perception::inference::ParseNetParam (const NetParameter &net_param, TensorDimsMap *tensor_dims_map, std::map< std::string, std::string > *tensor_modify_map, std::vector< LayerParameter > *order) |
bool | apollo::perception::inference::modify_pool_param (PoolingParameter *pool_param) |
bool | apollo::perception::inference::ParserConvParam (const ConvolutionParameter &conv, ConvParam *param) |
nvinfer1::DimsCHW | apollo::perception::inference::getCHW (const nvinfer1::Dims &d) |