tensorrtx
tensorrtx copied to clipboard
Implementation of popular deep learning networks with TensorRT network definition API
could you please the Mxnet and numpy version that I have to use? ``` File "/usr/local/lib/python3.8/dist-packages/mxnet/numpy/multiarray.py", line 47, in from .utils import _get_np_op File "/usr/local/lib/python3.8/dist-packages/mxnet/numpy/utils.py", line 37, in bool =...
## Env - GPU, 1660Ti, laptop - OS, win11 - Cuda 11.8 - TensorRT 8.6.1.6 ## About this repo - which branch/tag/commit are you using? yolov5-v7.0 - which model? yolov8n...
I am attempting to build a tensorrtx image of arm64, will you offer the dockerfile and source code of building the tensrrtx image. Thans a lot.
I try to convert yolov5.pt model to wts format and when I enter the python gen_wts.py -w best.pt -o best.wts I get this error: ``` {"payload":{"allShortcutsEnabled":false,"fileTree":{"yolov5":{"items":[{"name":"plugin","path":"yolov5/plugin","contentType":"directory"},{"name":"src","path":"yolov5/src","contentType":"directory"},{"name":"CMakeLists.txt","path":"yolov5/CMakeLists.txt","contentType":"file"},{"name":"README.md","path":"yolov5/README.md","contentType":"file"},{"name":"gen_wts.py","path":"yolov5/gen_wts.py","contentType":"file"},{"name":"images","path":"yolov5/images","contentType":"symlink_directory"},{"name":"yolov5_cls.cpp","path":"yolov5/yolov5_cls.cpp","contentType":"file"},{"name":"yolov5_cls_trt.py","path":"yolov5/yolov5_cls_trt.py","contentType":"file"},{"name":"yolov5_det.cpp","path":"yolov5/yolov5_det.cpp","contentType":"file"},{"name":"yolov5_det_cuda_python.py","path":"yolov5/yolov5_det_cuda_python.py","contentType":"file"},{"name":"yolov5_det_trt.py","path":"yolov5/yolov5_det_trt.py","contentType":"file"},{"name":"yolov5_seg.cpp","path":"yolov5/yolov5_seg.cpp","contentType":"file"},{"name":"yolov5_seg_trt.py","path":"yolov5/yolov5_seg_trt.py","contentType":"file"}],"totalCount":13},"":{"items":[{"name":".github","path":".github","contentType":"directory"},{"name":"alexnet","path":"alexnet","contentType":"directory"},{"name":"arcface","path":"arcface","contentType":"directory"},{"name":"centernet","path":"centernet","contentType":"directory"},{"name":"crnn","path":"crnn","contentType":"directory"},{"name":"dbnet","path":"dbnet","contentType":"directory"},{"name":"densenet","path":"densenet","contentType":"directory"},{"name":"detr","path":"detr","contentType":"directory"},{"name":"docker","path":"docker","contentType":"directory"},{"name":"efficientnet","path":"efficientnet","contentType":"directory"},{"name":"googlenet","path":"googlenet","contentType":"directory"},{"name":"hrnet","path":"hrnet","contentType":"directory"},{"name":"ibnnet","path":"ibnnet","contentType":"directory"},{"name":"inception","path":"inception","contentType":"directory"},{"name":"lenet","path":"lenet","contentType":"directory"},{"name":"lprnet","path":"lprnet","contentType":"directory"},{"name":"mlp","path":"mlp","contentType":"directory"},{"name":"mnasnet","path":"mnasnet","contentType":"directory"},{"name":"mobilenet","path":"mobilenet","contentType":"directory"},{"name":"psenet","path":"psenet","contentType":"directory"},{"name":"rcnn","path":"rcnn","contentType":"directory"},{"name":"real-esrgan","path":"real-esrgan","contentType":"directory"},{"name":"refinedet","path":"refinedet","contentType":"directory"},{"name":"repvgg","path":"repvgg","contentType":"directory"},{"name":"resnet","path":"resnet","contentType":"directory"},{"name":"retinaface","path":"retinaface","contentType":"directory"},{"name":"retinafaceAntiCov","path":"retinafaceAntiCov","contentType":"directory"},{"name":"scaled-yolov4","path":"scaled-yolov4","contentType":"directory"},{"name":"senet","path":"senet","contentType":"directory"},{"name":"shufflenetv2","path":"shufflenetv2","contentType":"directory"},{"name":"squeezenet","path":"squeezenet","contentType":"directory"},{"name":"superpoint","path":"superpoint","contentType":"directory"},{"name":"swin-transformer","path":"swin-transformer","contentType":"directory"},{"name":"tsm","path":"tsm","contentType":"directory"},{"name":"tutorials","path":"tutorials","contentType":"directory"},{"name":"ufld","path":"ufld","contentType":"directory"},{"name":"unet","path":"unet","contentType":"directory"},{"name":"vgg","path":"vgg","contentType":"directory"},{"name":"yolop","path":"yolop","contentType":"directory"},{"name":"yolov3-spp","path":"yolov3-spp","contentType":"directory"},{"name":"yolov3-tiny","path":"yolov3-tiny","contentType":"directory"},{"name":"yolov3","path":"yolov3","contentType":"directory"},{"name":"yolov4","path":"yolov4","contentType":"directory"},{"name":"yolov5","path":"yolov5","contentType":"directory"},{"name":"yolov7","path":"yolov7","contentType":"directory"},{"name":"yolov8","path":"yolov8","contentType":"directory"},{"name":".gitignore","path":".gitignore","contentType":"file"},{"name":"LICENSE","path":"LICENSE","contentType":"file"},{"name":"README.md","path":"README.md","contentType":"file"}],"totalCount":49}},"fileTreeProcessingTime":8.282295,"foldersToFetch":[],"reducedMotionEnabled":null,"repo":{"id":223904726,"defaultBranch":"master","name":"tensorrtx","ownerLogin":"wang-xinyu","currentUserCanPush":false,"isFork":false,"isEmpty":false,"createdAt":"2019-11-25T09:01:36.000Z","ownerAvatar":"https://avatars.githubusercontent.com/u/15235574?v=4","public":true,"private":false,"isOrgOwned":false},"symbolsExpanded":false,"treeExpanded":true,"refInfo":{"name":"master","listCacheKey":"v0:1698999982.0","canEdit":false,"refType":"branch","currentOid":"2df7dd7f9100fc04262e2a53fc53666eed28f84c"},"path":"yolov5/gen_wts.py","currentUser":null,"blob":{"rawLines":["import sys","import argparse","import os","import struct","import...
## Env - GPU 3090 - OS Ubuntu20.04 - CUDA 11.1 - CUDNN 8.1.1 - TensorRT 8.0 I have tried tensorrtx for both yolov5v7.0 and yolov5v6.0 ,and generated the yolov5s.wts...
您好, 是否考虑在yolov5中增加nms使用cuda加速推理?
## Env - GPU, 1080ti - OS, e.g. Ubuntu18.04 - Cuda version 11.1 - TensorRT 8.2.5.1 The reasoning of the C version is normal,However, the python version will report this...
## Env - GPU, e.g. V100, RTX2080, TX2, Xavier NX, Nano, etc. - OS, e.g. Ubuntu16.04, Win10, etc. - Cuda version - TensorRT version ## About this repo - which...
https://github.com/AILab-CVC/YOLO-World