search

TUI-NICR / ESANet

ESANet: Efficient RGB-D Semantic Segmentation for Indoor Scene Analysis
Other
239 stars 52 forks source link

cannot compute argmax and copy back to cpu when use Tensorrt #33

Open hlyf-xs opened 2 years ago

hlyf-xs commented 2 years ago

Hello, This is a great job! But I found a question that the program cannot compute argmax and copy back to cpu when I use tensorrt to speed up the model. So It's very slow beacuse use np.argmax()

MagicHly commented 2 years ago

Hello,I also found that it is too slow to use np.argmax() recently.How can I sovle this question?