官方源码tensorrtx-master\yolov5目录下提供很多cpp和py的案例,博主这里给大家提供一个yolov5_det_trt.py和yolov5_seg_trt.py的精简代码,代码中剔除了很多辅助性的功能函数。 主要功能就是读取一张本地图片,并进行目标预测,博主这里以车牌识别为主,提供工程源码 代码结构如下: 其中yolov5_det_trt.py和yolov5_seg...
Implementation of popular deep learning networks with TensorRT network definition API - Wblossom/tensorrtx
This branch is 22 commits ahead of, 212 commits behind wang-xinyu/tensorrtx:master.Folders and files Name Last commit message Last commit date parent directory .. CMakeLists.txt README.md calibrator.cpp calibrator.h configure.h gen_wts_refinedet.py logging.h refinedet.cpp utils.h...
inference
// 1. generate mnasnet.wts from [pytorchx/mnasnet](https://github.com/wang-xinyu/pytorchx/tree/master/mnasnet) // 2. put mnasnet.wts into tensorrtx/mnasnet // 3. build and run cd tensorrtx/mnasnet mkdir build cd build cmake .. make sudo ./mnasnet -s // serialize model to ...
master tensorrtx/dbnet/ Go to file This branch is 17 commits behind wang-xinyu:master. Pull request Compare Latest commit Git stats History Files Type Name Latest commit message Commit time . . CMakeLists.txt README.md common.hpp dbnet.cpp logging.h utils.h ...
upczww/tensorrtxPublic forked fromwang-xinyu/tensorrtx NotificationsYou must be signed in to change notification settings Fork0 Star1 Code Pull requests Actions Projects Security Insights Additional navigation options Files master .github alexnet
Implementation of popular deep learning networks with TensorRT network definition API - wuzhishiwo/tensorrtx
Implementation of popular deep learning networks with TensorRT network definition API - tensorrtx/yolov9 at master · wang-xinyu/tensorrtx
git clone https://github.com/wang-xinyu/tensorrtx.git git clone https://github.com/xuexingyu24/License_Plate_Detection_Pytorch.git // copy tensorrtx/LRPnet/gen_wts.py to License_Plate_Detection_Pytorch/ // go to License_Plate_Detection_Pytorch/ python genwts.py // a file 'LPRnet.wts'...