YOLOv3-Torch2TRT
YOLOv3-Torch2TRT copied to clipboard
DocF
Convert YOLOv3 and YOLOv3-tiny (PyTorch version) into TensorRT models.
when I run the detect.py and set **TensorRT = True** It went error. > Traceback (most recent call last): > File "/home/tx2wry/PycharmProjects/yolov3_torch2trt/detect.py", line 190, in > detections = non_max_suppression(detections, opt.conf_thres,...
Hi, There's no detection on the images after I set tensorrt flag = True and Half flag = True, I'd like to ask does it work on your original work?
(openpose) root@ai:~/YOLOv3-Torch2TRT# python3 detect.py Traceback (most recent call last): File "detect.py", line 113, in model_backbone.load_darknet_weights(opt.weights_path) File "/root/YOLOv3-Torch2TRT/models.py", line 365, in load_darknet_weights conv_w = torch.from_numpy(weights[ptr : ptr + num_w]).view_as(conv_layer.weight) RuntimeError: shape...
When I run this demo on my Jetson Nano with TensorRT mode,the out put contaions NONE detection!
[TensorRT] ERROR: (Unnamed Layer* 203) [Concatenation]: all concat input tensors must have the same number of dimensions, but mismatch at input 1. Input 0 shape: [1,256,26,26], Input 1 shape: [512,26,26]...
HI, thanks for your repo, the performance seems to be amazing ! I'm trying to run it on my Jetson Xavier but I'm not able to find out why I...
The model doesn't detect any object! I printed the detections tensor before and after the Non Maximum Suppression and here are the outputs: ``` Before tensor([[[ 0., 0., 0., ...,...
