haotian/duan8
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
317 Commits 2 Branches 0 Tags 7.8 MiB
46dccf11ea
Commit Graph

317 Commits

This Branch
This Branch
All Branches
Author SHA1 Message Date
Dominic D
46dccf11ea
Update YOLOv3-tiny for TRT8 (#1064) 
* Update YOLOv3-tiny for TRT8
Changes from #967

* Also apply fixes to utils

* Remove redundant macro definition in logging.g
 2022-08-02 13:33:13 +08:00
Wang Xinyu
0f944b8975
Update README.md 2022-07-06 11:35:17 +08:00
Xiang Shin Wuu
fdd136466f
SuperPoint TensorRT implementation (#1034) 
* adding tensorrt implementation of superpoint network.

* inference output result & todo list added to README
 2022-07-06 11:19:10 +08:00
Egor Tolmachev
dbdad6846c
Fix comparison with args in mode check. (#1019) 
@wang-xinyu @wdhao  Before we had assignment, not comparison.(somebody forgot "=" but not only)
 2022-06-27 13:51:24 +08:00
haku
e418cb9e66
add dockerfile trt8_cuda11 (#1010) 
* add dockerfile trt8_cuda11

* update docker readme
 2022-06-11 11:51:41 +08:00
HaiyangP
6636cc26fd
Modifications in rcnn.cpp for more applicable outputs of instance segmentation. (#946) 
* Support to output the instance segmentation results on an original input image, rather than on a padded one.

* modifications according to freedenS's suggestions

* modification for padding

* merge preprocessImg and preprocessImg_ and delete all the debug codes.

* bug of reference passing

* make the vars consistent with the previous rcnn.cpp and update the README

Co-authored-by: benkang wang <benkang.wang@uisee.com>
 2022-06-06 11:03:28 +08:00
Wang Xinyu
e2f270a0ce
Update README.md 2022-05-28 07:59:05 +08:00
Wang Xinyu
6c59b8bc26
Update README.md 2022-05-27 08:44:34 +08:00
Wang Xinyu
b91b15d3e7
Update README.md 2022-05-27 08:38:03 +08:00
tripleMu
bfe0b55406
Replace pycuda api with cuda-python (#1002) 
* Replace cuda-python api of pycuda

* Remove my test batch=2
 2022-05-26 09:37:06 +08:00
Wang Xinyu
f80f010ac4
Update README.md 2022-05-23 10:44:07 +08:00
Wang Xinyu
3705a1a095
fix typo 2022-05-23 10:32:59 +08:00
Wang Xinyu
8a472a1c6d
Update README.md 2022-05-23 10:30:57 +08:00
yhpark
e0c5243a53
added real-esrgan (#999) 
* added real-esrgan

* deleted sample image & modified README

* tab to space
 2022-05-23 10:14:05 +08:00
Vijaysrinivas Rajagopal
42b4f94527
YOLOv3 TRT8 support 
* yolov3 trt8 fix

* move the if,else to macro header like yolov5

* reset imagesize to 608 and other stuff
 2022-05-19 14:28:54 +08:00
Wang Xinyu
6ddff90025
Update README.md 2022-03-15 13:02:50 +08:00
Wang Xinyu
e7ec22d8b8
add swin-transformer (#934) 
* add swin-transformer

* change dir

* reformat

* fix gen wts and polish readme

Co-authored-by: wdhao <873665182@qq.com>
 2022-03-15 12:10:49 +08:00
tripleMu
3f31d76ac4
Fix bug (#931) 
Co-authored-by: tripleMu <92794867+q3394101@users.noreply.github.com>
 2022-03-11 13:45:56 +08:00
AdanWang
70059d1802
modify load model for ema (#929) 
Co-authored-by: 王光强 <wangguangqiang@cethik.com>
 2022-03-11 13:44:17 +08:00
LW-SCU
c2a07ff7ac
Fix wrong use of copyMakeBorder (#883) 
Co-authored-by: lwjustsoso <351048599@qq.com>
 2022-01-28 11:34:59 +08:00
Subtain Malik
f2ac76df5a
added MLP with proper documentation (#874) 
* [MLP]: create using core TensortRT python APIs

* [MLP]: README.md added with chart

* [MLP]: C++ TensorRT APIs added

* [MLP]: Updated Docs

* [MLP]: basic .wts added with APIs

* [MLP]: Updated minor details

* [MLP]: fix
 2022-01-10 11:34:23 +08:00
Vijaysrinivas Rajagopal
e6b22917e6
YOLOv3 TRT Python (#873) 
* change yolov5_trt to fit yolov3 output

* change ref from yolov5 to v3
 2022-01-07 11:09:48 +08:00
xupengao
5f1b048a29
fix bug (#808) 
* Add files via upload

* check_file

* remove invalid data

* remove data

* Optimization and modification

add all  yolo model

* fix bug

* fix bug

Co-authored-by: Wang Xinyu <wangxinyu_es@163.com>
 2021-11-09 13:53:20 +08:00
Silmeria112
befdf6ee64
update gen_wts.py for yolov5 to support custom model with different anchor from official model(#788) 2021-10-29 11:10:12 +08:00
Mr.Li
4697890416
Fix error when using yolov5-v6 custom mode (#782) 2021-10-26 13:25:30 +08:00
Jason
4686140d25
yolov5: fix bbox inaccurate issue (#779) 
Co-authored-by: chenjiansong <chenjiansong@trunk.tech>
 2021-10-21 22:24:36 +08:00
liuqi
2fef5f0ef2
yolov5: read input images only once (#776) 
you only read once。 把预处理图片缓存下来，画框时不必读一次，一张图片后处理时间少10ms。
 2021-10-21 11:14:24 +08:00
wang-xinyu
e785f51d10 update readme 2021-10-20 05:31:24 +00:00
wang-xinyu
542ab93fd3 reformat preprocess code of yolov5 2021-10-19 11:49:37 +00:00
liuqi
e7a5f0acbc
preprocess image in GPU for yolov5 (#770) 
* Update CMakeLists.txt

add nvcc compile option ;
add source file : preprocess.cu

* Create preprocess.h

* Update README.md

Add some configurations

* Update README.md

* Update CMakeLists.txt

* Update CMakeLists.txt

* Create preprocess.cu

* Update yolov5.cpp

Preprocess the image in GPU;
Use pinned memory to speed up the copy;
Only "cudamemcpy" once in function "doInference";

* Update preprocess.h

* Update CMakeLists.txt

* Update preprocess.cu

* Update yolov5.cpp

* Update preprocess.cu

* Update preprocess.cu

* Update preprocess.cu

* Update preprocess.cu

* Update preprocess.cu

* Update preprocess.cu

* Update preprocess.cu
 2021-10-19 18:56:47 +08:00
wang-xinyu
ae329092a1 update yolov5 readme 2021-10-18 07:43:48 +00:00
Wang Xinyu
8aa36a36f6
Update common.hpp 2021-10-18 13:51:44 +08:00
xupengao
a01fae5022
yolov6 6.0 Modification and optimization (#764) 
* Add files via upload

* check_file

* remove invalid data

* remove data

* Optimization and modification

add all  yolo model

Co-authored-by: Wang Xinyu <wangxinyu_es@163.com>
 2021-10-18 13:47:08 +08:00
xupengao
bcc01facfa
yolov5 6.0 (#763) 
* Add files via upload

* check_file

* remove invalid data

* remove data
 2021-10-18 11:20:23 +08:00
wang-xinyu
791c8a4484 update yolov5 readme 2021-10-13 12:15:10 +00:00
JumpPandaer
c70a338827
fix a bug with maskrcnn (#758) 2021-10-13 19:04:39 +08:00
liufqing
7c1a145c34
add resnet34 (#741) 
* hello

* add resnet34
 2021-09-30 15:18:04 +08:00
Armassarion
9fefbed77f
Fix a potential bug in detection number reset between each inference (#733) 
Co-authored-by: Ksander <ksander.ejjaaouani@digeiz.com>
 2021-09-22 19:22:28 +08:00
HaiyangP
f7d224f9e4
Add the coordinate clipping op before nms (#704) 
* add the coordinate clipping op before nms

* modified the pytorch related codes

* modified the pytorch related codes
 2021-09-02 19:35:30 +08:00
Jian Han Lim
0e254088a9
Update retinaface_trt.py, preprocess_image code (#703) 
Proposed solution to solve this issue #702
 2021-09-01 14:29:55 +08:00
Wang Xinyu
35a3301bf7
Update README.md 2021-08-31 19:10:16 +08:00
FamousDirector
9f9109fdc9
give retinaface TRT8 support (#698) 
* give retinaface TRT8 support

* add macros.h to retinaface conversion

* add macros.h to retinaface conversion
 2021-08-31 19:05:43 +08:00
HaiyangP
7c4476f23c
A pure numpy version for yolov5_trt.py (#700) 
* This code provides a pure numpy manner to write nms operation in yolov5_trt.py. I think it can offer simple and convenient TensorRT experience for yolov5

* delete torch related codes
 2021-08-31 11:32:34 +08:00
wang-xinyu
32049c61d1 update yolov5 readme 2021-08-30 05:52:02 +00:00
Cristi Fati
bbbe455a45
Improvements (generate .wts file) (#691) 
* Improve output (wts) file name generation

* Named arguments

* Minor: empty out argument
 2021-08-30 11:14:25 +08:00
HaiyangP
bd0d13f610
add the tensorrt python api for hrnet (#692) 2021-08-27 20:32:54 +08:00
JumpPandaer
ec20cead6f
guide for measuring performance (#676) 2021-08-17 10:50:43 +08:00
JumpPandaer
ccdf0e6316
update README (#673) 
add latency

improve nms
 2021-08-16 10:59:57 +08:00
wang-xinyu
a34527a48d create macros.h for yolov5 2021-08-05 11:57:48 +00:00
BaofengZan
55ad6259ce
Fix compilation problems on Windows (#644) 2021-07-30 10:57:29 +08:00