Skip to content

sugamza95/2020-JETSON-NX-TERM-PRJ

Repository files navigation

This may not be run in your environment other than our teammates.

  1. https://github.com/NVIDIA-AI-IOT/trt_pose Follow the example int the website above and copy all directories and files in tasks folder to tensorRT-yolo/tensorrt_demos (in your case run test.py)

  2. run this command in tensorrt_demos

... 
$ cd tensorrt_demos git clone https://github.com/sugamza95/2020-JETSON-NX-TERM-PRJ.git
  1. copy yolov4-tiny-416.cfg, yolov4-tiny-416.weights, yolov4-tiny-416.onnx, yolov4-tiny-416.trt in yolo of my github to your tensorrt_demos/yolo directory
$ cd 2020-JETSON-NX-TERM-PRJ 
$ cd yolo 
$ cp yolov4-tiny-416.cfg ../../yolo 
$ cp yolov4-tiny-416.weights ../../yolo 
......

4.copy trt_yolo_with_centernet.py to tensorrt_demos directory

$ cp trt_yolo_with_centernet.py .. 
  1. cp yolo_with_plugins.py in the utils folder in my project directory to your /tensorrt_demos/utils folder
$ cd utils 
$ cp yolo_with_plugins.py ../../ 

5-1(optional). copy resnet18_baseline_att_224x224_A_epoch_249.py, resnet18_baseline_att_224x224_A_epoch_249_trt.py to tensorrt_demos

$ cd .. 
$ cp resnet18_baseline_att_224x224_A_epoch_249.py .. 
$ cp resnet18_baseline_att_224x224_A_epoch_249_trt.py .. 

6.change directories in trt_yolo_with_centernet.py ex)

out_video = cv2.VideoWriter('/home/ee201511281/street_2_term_prj.mp4', fourcc, 30.0 , (1280, 720))

7.run this command

$ python3 trt_yolo_with_centernet.py --video <video file> -m yolov4-tiny-416 

If you want to use your custom weights, cfg files refer to this github https://github.com/jkjung-avt/tensorrt_demos

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages