Commit Graph

11 Commits

Author SHA1 Message Date
Aymeric Dujardin
b7a754e2de
Fix for TensorRT 8 support (#540) 2021-07-05 12:02:56 +08:00
wang-xinyu
046dd5b417 yolov5 support s6/m6/l6/x6 models 2021-04-25 12:17:15 +00:00
Pavel Mikhadzionak
e72d9db48b
Fix synchronization error (#485) 2021-04-16 10:37:10 +08:00
wang-xinyu
1a82822474 fix and optimize 2021-01-29 16:20:48 +08:00
wang-xinyu
95b9b3bc9e yolov5 v4.0 int8 2021-01-24 13:58:06 +08:00
BaofengZan
55e604e6f7 Fix bug : when the width and height are different, the result is incorrect (#235) 2020-11-06 23:31:42 +08:00
wang-xinyu
08126b175d fix yolov5 2020-11-06 23:27:30 +08:00
BaofengZan
8bdbb47dfa 从wts文件中读取anhor信息,并传入yololayer plugin (#215)
* 修改图片地址为URL

* get Anchor values from wts

* get anchor values form wts

* get anchor values from wts

* get anchor values from wts

* 1 using space 2 release heap mem

* 1 Using 4 space
2 Delete Chinese comment
3 Load all variables
4 Release heap mem

* 1 Using 4 space
2 Delete Chinese comment
3 Load all variables
4 Release heap mem

* 1. 函数模块化
2. 格式化代码
3. 同步为最新的yolov5代码
2020-11-06 23:27:30 +08:00
wang-xinyu
efb809c043 optimize gpu buffer 2020-11-06 22:54:15 +08:00
wang-xinyu
271b152c30 yolov5s batchsize 2020-06-22 19:02:30 +08:00
wang-xinyu
6ee4c62ff0 implement yolov5s 2020-06-22 13:48:17 +08:00