tjuskyzhang / Scaled-YOLOv4-TensorRT

Got 100fps on TX2. Got 500fps on GeForce GTX 1660 Ti. If the project is useful to you, please Star it.
178 stars 41 forks source link
pytorch scaled-yolov4 scaled-yolov4-tenosrrt tensorrt tx2 yolo yolov4-csp yolov4-csp-tensorrt yolov4-large yolov4-large-tensorrt yolov4-tiny yolov4-tiny-tensorrt

README

Scaled-YOLOv4-TensorRT:

Implement yolov4-tiny-tensorrt, yolov4-csp-tensorrt, yolov4-large-tensorrt(p5, p6, p7) layer by layer using TensorRT API.