Test yolov3-trt on jetson nano
- Python 3
- OpenCV
- PyTorch
- Onnx 1.4.1
- Tensorrt
mkdir cfg
cd cfg
原链接失效百度网盘地址
链接:https://pan.baidu.com/s/1CtWGfJQ2PWl6kLoMsH7Yhg
提取码:gm4v
mkdir weights
cd weights
wget https://pjreddie.com/media/files/yolov3.weights
Inorder to Run the model in Pytorch or creating Onnx / Tensorrt File for different Input image Sizes ( 416, 608, 960 etc), you need to edit the Batch Size and Input image size in the config file - net info section.
batch=1
width=416
height=416
python3 create_onnx.py --reso 416
python3 onnx_to_tensorrt.py --model yolov3-416
Creating the Tensorrt engine takes some time. So have some patience.
Test the YOLOv3 TensorRT engine with the "dog.jpg" image.(jetson nano run 2.55 FPS,jetson xavier nx run 11.59 FPS)
python3 trt_yolov3.py --model yolov3-416
--image --filename ${HOME}/Pictures/dog.jpg
python3 trt_yolov3.py --usb --vid 0 --width 1280 --height 720 (or 640x480)
python3 eval_yolov3.py --model yolov3-416