tensorrtx icon indicating copy to clipboard operation
tensorrtx copied to clipboard

Implementation of popular deep learning networks with TensorRT network definition API

Results 156 tensorrtx issues
Sort by recently updated
recently updated
newest added

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