The trt_yolov3 from royinx

trt_yolov3's Introduction

pull container 19.02 py2 / 19.12 py3

docker pull nvcr.io/nvidia/tensorrt:19.02-py2
docker pull nvcr.io/nvidia/tensorrt:19.12-py3

Run conatainer from 1-Step mini build

docker build -t trt_yolov3 .
docker run --rm -it --privileged --runtime=nvidia trt_yolov3 bash
python3 onnx_to_tensorrt.py --build --vram 8 --max_batch_size 64 -p fp16

Run container from scratch

--privileged --runtime=nvidia

(19.02 for yolo2onnx , TRT 5.0.2 , onnx 1.3.0 or 1.2.3 )

docker run --rm -it --privileged --runtime=nvidia -v ~/Desktop/python:/sharefolder nvcr.io/nvidia/tensorrt:19.02-py2 bash
pip install wget onnx==1.3.0 
cd tensorrt/samples/python/yolov3_onnx/ && python yolov3_to_onnx.py
cp yolov3.onnx /sharefolder/

(19.12 for onnx2trt , TRT 6.0.1.8 , onnx 1.7.0)

docker run --rm -it --privileged --runtime=nvidia -v ~/Desktop/python:/sharefolder nvcr.io/nvidia/tensorrt:19.12-py3 bash
pip3 install wget onnx scipy line_profiler
git clone https://github.com/royinx/trt_yolov3.git
cp /sharefolder/yolov3.onnx trt_yolov3/ && cd trt_yolov3
python onnx_to_tensorrt.py --build --vram 8 --max_batch_size 64 -p fp16

Others

Debug core dump (inside container)

commit new image and run at a new container

install gdb apt-get update && apt-get -y install gdb python2.7-dbg

$ cd

$ gdb python

(gdb) run XXX.py
(gdb) generate-core-file
(gdb) py-bt
(gdb) q

trt_yolov3's People

Contributors

Stargazers

Watchers

trt_yolov3's Issues

python yolov3_to_onnx.py for error.

Hi,@royinx
Tried to python yolov3_to_onnx.py

Layer of type yolo not supported, skipping ONNX node generation.
Layer of type yolo not supported, skipping ONNX node generation.
Layer of type yolo not supported, skipping ONNX node generation.
graph YOLOv3-608 (
%000_net[FLOAT, 64x3x608x608]
) initializers (
%001_convolutional_bn_scale[FLOAT, 32]
%001_convolutional_bn_bias[FLOAT, 32]
%001_convolutional_bn_mean[FLOAT, 32]
my python2.7 ,onnx==1.4.1, TensorRT-6.0.1.5 cuda-10.0 Ubuntu18.04, yolov3.weights ,yolov3.cfg

Thanks!

Recommend Projects