HaiyangP
877bb24389
Optional nms methods supporting for rcnn ( #1274 )
...
* 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
* support different nms methods
* cxx11
* add contributors and delete interpretations
* update README
* fix some problems
* fix the bugs
---------
Co-authored-by: benkang wang <benkang.wang@uisee.com>
2023-04-19 17:00:14 +08:00
tsieyy
30de4562bc
Adapt ufld model to tensorrt8 ( #1288 )
...
* make ufld adapt tensorrt8
* make ufld adapt tensorrt8 and tensorrt7
2023-04-19 12:45:43 +08:00
nengwp
f2521a5ffa
RCNN upgrade to support TensorRT 7&8 ( #1256 )
...
* rcnn upgrade to support TensorRT 8.
* Update RpnDecodePlugin.h
Remove Chinese
* Update rcnn.cpp
Remove Chinese
* Update backbone.hpp
Remove Chinese
* Update backbone.hpp
* Update rcnn.cpp
* Update rcnn.cpp
* Update rcnn.cpp
* Update MaskRcnnInferencePlugin.h
* rcnn upgrade to support TensorRT 8.x
* rcnn upgrade to support TensorRT 8.x
* Update macros.h
* Update README.md
---------
Co-authored-by: nengwp <nengwp@github.ai>
Co-authored-by: Wang Xinyu <shaywxy@gmail.com>
2023-03-07 19:26:53 +08:00
nengwp
ef22b1d03b
UNet support TensorRT 8.x ( #1245 )
...
* TensorRT 8.x supported
* Unet TensorRT 8.x supported, add Contributors nengwp
---------
Co-authored-by: nengwp <nengwp@github.ai>
2023-02-28 10:23:40 +08:00
lishanxiuxiu
f92dcf43dc
Fix dead link in yolov7 readme ( #1222 )
2023-02-02 12:50:58 +08:00
wang-xinyu
c66b8b7e2c
update readme
2023-01-17 15:58:37 +08:00
Wang Xinyu
208032970c
Refactor yolov5 ( #1211 )
...
* refactor pre and post process
* move model, refactor plugin
* refactor yolov5 cls
* refactor yolov5 seg
* fix space
* rename samples
* update readme
2023-01-17 15:42:47 +08:00
Wang Xinyu
57ba75f9a4
refactor yolov5_det.cpp ( #1209 )
2023-01-16 19:46:59 +08:00
mantuoluozk
f334a1aa38
yolop TRT8 support ( #1202 )
...
* yolop TRT8 support
* yolop TRT8 support
2023-01-11 14:05:42 +08:00
Prince Patel
772d353399
Draw label text for yolov5_seg ( #1197 )
...
* convert class id to class name + added confidence + different colors for class ids already present + background box to match with yolov5 repo + added coco.txt
* Required changes: removed coco.txt and *.jpg files, updated readme
* Resolving comments
* Update utils.h
* Update yolov5_seg.cpp
* Update README.md
Co-authored-by: Wang Xinyu <shaywxy@gmail.com>
2023-01-11 10:54:16 +08:00
Rex
7b79de466c
fix a problem in warmUp ( #1194 )
...
Int process of warmUp，the pred is None, resulting in the error of "masks = self.sigmoid((result_proto_coef @ result_proto_masks.astype(np.float32).reshape(c, -1))).reshape(-1, mh, mw)"
2023-01-03 15:58:16 +08:00
Rex
3c0f62e82f
delete useless code ( #1188 )
2022-12-23 16:54:55 +08:00
Rex
7caa7e0dd3
Yolov5 seg python inference ( #1185 )
...
* add yolov5_seg_trt.py
* complete yolov5-instence seg
* fix yolov5-seg
* fix_yolov5-seg
* fix_yolov5-seg
* fix code style
* fix coding style
Co-authored-by: Wang Xinyu <shaywxy@gmail.com>
2022-12-22 20:33:40 +08:00
Rex
d6a0b784c4
fix_yolov5 ( #1181 )
...
* fix_yolov5
* add fix
* add Rex
2022-12-21 19:40:23 +08:00
wang-xinyu
aac418e1f0
yolov5/7: fix batch processing bug
2022-12-21 11:06:35 +08:00
wang-xinyu
ebc54830ec
refactor yolov5 dir
2022-12-20 13:42:59 +08:00
Wang Xinyu
fbad2261dc
yolov5: upgrade to v7.0 and support instance segmentation ( #1177 )
...
* add seg
* yolov5 seg
* update wts and readme
* add todo
* update readme
2022-12-18 23:15:07 +08:00
wang-xinyu
165b0a428c
yolov5: rm imagenet_classes.txt
2022-12-16 22:02:10 +08:00
wang-xinyu
712a50bc08
yolov5: use downloaded imagenet lables
2022-12-16 21:40:19 +08:00
uyolo
90b2f1e445
yolov5: load strides from .wts file ( #1172 )
...
* update load strides from model
* load strides from model file
2022-12-14 21:08:45 +08:00
Wang Xinyu
cdf45858c6
update introduction
2022-12-13 23:31:44 +08:00
Wang Xinyu
cb9efbdf21
unet: fix bug and coding style ( #1171 )
...
* unet: fix bug
* update readme
* remove useless code
2022-12-13 22:37:41 +08:00
East-Face
f98182e1b1
UNet: adapted to Pytorch-UNet v3.0 ( #1169 )
...
* add deconvlution
* Delete unet/samples directory
* Update README.md
Co-authored-by: liuxiangd <liuxiangd@mailyunda.com>
2022-12-13 17:22:07 +08:00
Zeyad Mansour
edecefe4c7
Update main.cpp ( #1167 )
2022-12-09 21:04:37 +08:00
wang-xinyu
ad7e3052f4
fix yolov7x bug
2022-11-29 20:44:34 +08:00
Wang Xinyu
816eef5194
Refactor yolov7 ( #1155 )
...
* separate model and block
* move thresh to config
* use trt8 api
* move batchsize to config
* update readme
* refactor batch infer logic
* move max input image size to config
2022-11-27 23:33:34 +08:00
Wang Xinyu
4a01957294
separate model and block ( #1154 )
2022-11-26 23:15:24 +08:00
Wang Xinyu
fcdf274119
Refactor yolov7 ( #1153 )
...
* use cmake 3.10
* refactor yolov7 code
* refactor yolov7 parse_args
* remove anchor definition
2022-11-26 20:41:43 +08:00
wang-xinyu
9071fa2d2b
unet fix coding style
2022-11-25 21:42:01 +08:00
Wang Xinyu
e061888ed2
Update unet readme
2022-11-25 21:29:46 +08:00
wang-xinyu
e94e4ed70f
unet support trt8
2022-11-25 21:13:49 +08:00
jinlj
1512aa73ae
Add yolop_trt.py ( #1131 )
...
* add simplified Logger class
* add modified yolop tensorrtx implementation
* add gen_wts.py
* add CMakeLists.txt
* add yolop tensorrtx README.md
* fix yolop spelling
* simplify yolop code
* fix yolop README.md
* Update README.md
* Add yolop_trt.py
* Update yolop README.md
* simplify yolop_trt.py
Co-authored-by: aliceint(zhanglj) <2464313547@qq.com>
Co-authored-by: Wang Xinyu <wangxinyu_es@163.com>
2022-11-01 19:21:12 +08:00
Wang Xinyu
9243edf59e
Update README.md
2022-10-26 20:26:54 +08:00
Wang Xinyu
9253d58e7a
Update README.md
2022-10-26 15:18:08 +08:00
jinlj
208b2fcf39
Feature: add modified yolop tensorrt implementation ( #1129 )
...
* add simplified Logger class
* add modified yolop tensorrtx implementation
* add gen_wts.py
* add CMakeLists.txt
* add yolop tensorrtx README.md
* fix yolop spelling
* simplify yolop code
* fix yolop README.md
Co-authored-by: aliceint(zhanglj) <2464313547@qq.com>
2022-10-26 15:04:27 +08:00
wang-xinyu
0c78095dd6
lenet5: support trt8
2022-10-09 11:30:31 +08:00
Wang Xinyu
87f897bced
Merge pull request #1117 from LiberiFatali/master
...
Attach the decode kernel to the same CUDA stream
2022-10-09 10:54:12 +08:00
LiberiFatali
569cc4a438
Attach the decode kernel to the same CUDA stream
...
fix to load engine file in DeepStream 6.1.1
2022-10-08 21:41:35 +07:00
wang-xinyu
dfc437daa0
fix coding style
2022-09-29 18:48:29 +08:00
wang-xinyu
1c65f082e0
fix coding style
2022-09-23 16:03:40 +08:00
wang-xinyu
4079560f95
fix code style
2022-09-23 15:43:36 +08:00
wang-xinyu
ccbf1170b9
yolov7 samples soft link
2022-09-23 15:34:08 +08:00
lindsayshuo
dc1e279516
update gen_wts.py and some yolov5's names ( #1109 )
...
* update gen_wts.py and some yolov5's names
* Update gen_wts.py
2022-09-22 11:03:37 +08:00
Wang Xinyu
ca5e9425f9
Update README.md
2022-09-19 16:02:00 +08:00
Wang Xinyu
fb674add10
Update README.md
2022-09-19 15:55:49 +08:00
lindsayshuo
64da26ef5d
yolov7 v0.1 ( #1105 )
...
* yolov7 v0.1
* Update README.md
* Update README.md
* Update README.md
* Update README.md
* Update README.md
* Update README.md
* Update README.md
* Update yololayer.h
* Update yololayer.h
* Delete yolov7.engine
* Delete yolov7-tiny.engine
* Delete yolov7/output directory
* Delete dirent.h
* Update README.md
* Update README.md
* Update README.md
* Update README.md
2022-09-19 15:49:39 +08:00
Wang Xinyu
d988519e73
Update README.md
2022-09-07 16:53:57 +08:00
wang-xinyu
10fd212890
update readme
2022-09-07 15:36:03 +08:00
wang-xinyu
b8d8c8ab92
yolov5 cls cpp postprocessing
2022-09-07 14:58:12 +08:00
wang-xinyu
2c0c7539e3
yolov5 cls cpp inference
2022-09-06 20:35:26 +08:00
