朝圣者项目：torch2trt，将pytorch模型快速转换为TensorRT引擎。-C/C++开发 （296个子文件） 
modulatedDeformableConvPlugin.cpp 19KB deformableConvPlugin.cpp 18KB gridAnchorDynamicPlugin.cpp 16KB groupNormPlugin.cpp 13KB delta2bboxPlugin.cpp 12KB meshGridPlugin.cpp 12KB layerNormPlugin.cpp 12KB batchedNMSPlugin.cpp 11KB roiExtractorPlugin.cpp 11KB carafeFeatureReassemblePlugin.cpp 11KB roiPoolPlugin.cpp 9KB deformablePoolPlugin.cpp 9KB deform_conv_cuda.cpp 9KB gridSamplePlugin.cpp 9KB repeatDimsPlugin.cpp 8KB adaptivePoolPlugin.cpp 8KB exViewPlugin.cpp 8KB torchCumMaxMinPlugin.cpp 7KB torchGatherPlugin.cpp 7KB torchFlipPlugin.cpp 6KB torchNMSPlugin.cpp 6KB torchCumPlugin.cpp 6KB interpolate.cpp 6KB batchedNMSInference.cpp 6KB expressionParser.cpp 4KB NvDsInferParseMmdet.cpp 4KB nmsHelper.cpp 2KB kernel.cpp 1KB InferPlugin.cpp 1KB logger.cpp 1KB grid_sample.cu 23KB allClassNMS.cu 13KB deform_conv_cuda_kernel.cu 13KB roi_extractor_kernel.cu 10KB roi_pool_kernel.cu 10KB torch_nms.cu 10KB adaptive_pool.cu 8KB carafe_cuda_kernel.cu 8KB sortScoresPerClass.cu 8KB gatherNMSOutputs.cu 8KB delta2bbox.cu 6KB tensorMeanVar.cu 5KB torch_cum_maxmin.cu 5KB common.cu 5KB sortScoresPerImage.cu 4KB permuteData.cu 4KB torch_cum.cu 3KB torch_gather.cu 3KB torch_flip.cu 3KB repeatDims.cu 2KB mesh_grid.cu 2KB group_norm_kernel.cu 2KB memcpyPermute.cu 2KB layer_norm_kernel.cu 2KB deform_roi_pool.cu 2KB grid_anchor_dynamic.cu 2KB reduceUtils.cuh 8KB deform_roi_pool_cuda_kernel.cuh 4KB .gitignore 185B .gitignore 9B .gitignore 6B common.h 28KB common.h 27KB logging.h 16KB plugin.h 12KB kernel.h 10KB modulatedDeformableConvPlugin.h 5KB deformableConvPlugin.h 5KB gridAnchorDynamicPlugin.h 4KB groupNormPlugin.h 4KB layerNormPlugin.h 4KB batchedNMSPlugin.h 4KB batchedNMSPlugin.h 4KB roiExtractorPlugin.h 4KB delta2bboxPlugin.h 4KB deformablePoolPlugin.h 4KB meshGridPlugin.h 4KB roiPoolPlugin.h 4KB carafeFeatureReassemblePlugin.h 4KB adaptivePoolPlugin.h 4KB gridSamplePlugin.h 4KB exViewPlugin.h 4KB torchCumMaxMinPlugin.h 4KB torchCumPlugin.h 4KB torchGatherPlugin.h 4KB torchFlipPlugin.h 4KB torchNMSPlugin.h 4KB repeatDimsPlugin.h 4KB amirCommon.h 3KB bboxUtils.h 2KB bboxUtils.h 2KB cuda_util.h 2KB batchedNMSInference.h 1KB gatherNMSOutputs.h 1KB cub_helper.h 1KB deform_conv_cuda.h 1012B logger.h 969B nmsUtils.h 939B carafe_cuda.h 685B roi_extractor.h 683B共 296 条
- 1
- 2
- 3
资源评论
