hrnet classification trt 加速

This commit is contained in:
BaofengZan 2020-09-21 12:42:54 +08:00 committed by wang-xinyu
parent a0535c4876
commit 3d2228d7a9

View File

@ -18,7 +18,7 @@ The Pytorch implementation is [HRNet-Image-Classification](https://github.com/HR
cd build
cmake ..
make
sudo ./dbnet -s // serialize model to plan file i.e. 'DBNet.engine'
sudo ./dbnet -s // serialize model to plan file i.e. 'hrnet.engine'
sudo ./dbnet -d ../samples // deserialize plan file and run inference, the images in samples will be processed.
```