mmdetection-to-tensorrt

mmdetection-to-tensorrt copied to clipboard

error:batchedNMSPlugin.cpp

7

hi,I met the problem: #assertion/amirstan_plugin/src/plugin/batchedNMSPlugin/batchedNMSPlugin.cpp,143 Aborted (core dumped)  **enviroment:** - OS: [Ubuntu] - python_version: [3.6] - pytorch_version: [1.6] - cuda_version: [cuda-10.2] - cudnn_version: [7.6.5] - mmdetection_version: [2.4] Looking forward...

lexiqi

performs well in NVIDIA agx platform，and some questions about batch inference

3

Thanks for your help, I have tested this tool in NVIDIA agx platform, we get a nearly 3X speed up compared using mmdetection directly.That's an amazing result! And I have...

heboyong

inference

20

Can produce trt but there will be a prompt "Warning: Encountered known unsupported method torch.Tensor.new_zeros" But during the test, it prompted a problem. "[TensorRT] ERROR: (Unnamed Layer* 204) [ElementWise]: dimensions...

manhongnie

Thanks for your great work, and there are some questions for help

1

This is a very good and helpful work for me ,thank you.But I have two questions about your work: 1.For 'mmdetection-to-tensorrt',is the 'amirstan_plugin' tool must needed? Because one of the...

heboyong

skeleton or stgcn module

2

hi~how long will the skeleton or st-gcn module be added? will it take me long time to write a module supporting skeleton and st-gcn？I haven't written a program about tensorrt....

xn1997

教程

1

请问作者，有教程文档说明嘛？

python-bookworm

GPU Memory-Usage Control

1

Refering to the [issue in Project mmdetection-to-tensorrt](https://github.com/grimoire/mmdetection-to-tensorrt/issues/8#issue-683354899) Multiple batch for one inference was support now. cool ~ However, GPU Memory-Usage will be increased when using multiple batch ( GPU Memory...

cefengxu

Multiple batch for only one inference

4

how to process more than one images ( for example 2 or batch_size == 2 ) on one inference when I using mmdetection-to-tensorrt ?

cefengxu

First, I test a mmdet demo on https://github.com/open-mmlab/mmdetection, and it works. Then I convert the .pth model to .trt model use mmdet2trt, and it works as well. But when I...

zkccpro

Fix a bug when detection number=0. Fix a bug when loading ConcatDataset in config file.

chenxinfeng4