WuxinrongY
551be9d939
update gen_wts.py ( #1467 )
2024-03-26 13:30:00 +08:00
Wang Xinyu
f12bbe0359
fix code style ( #1460 )
2024-03-20 11:36:16 +08:00
Wang Xinyu
d3e427f73c
Update pre-commit.yml
2024-03-20 11:27:18 +08:00
Wang Xinyu
a5f84f6103
pre-commit check and contribution doc ( #1459 )
...
* remove blank
* Update pre-commit.yml
* Update pre-commit.yml
* code style
* add contribution.md
* add contribution.md
2024-03-20 11:13:45 +08:00
Wang Xinyu
1125a64d80
Update pre-commit.yml
2024-03-19 20:54:34 +08:00
Wang Xinyu
0752da558c
Update pre-commit.yml
2024-03-19 19:49:32 +08:00
Wang Xinyu
3fe9df9fa2
Create pre-commit.yml
2024-03-19 19:31:52 +08:00
wang-xinyu
f768d67fbc
add pre-commit action
2024-03-19 19:09:02 +08:00
wang-xinyu
f159a101bd
add pre-commit
2024-03-19 18:51:27 +08:00
Rex
e4da5a8367
detr添加对trt8的支持 ( #1457 )
...
* detr添加对trt8的支持
* 修改格式
* 修改格式
* 修改格式
* 修改格式
* Update detr.cpp
---------
Co-authored-by: Wang Xinyu <shaywxy@gmail.com>
2024-03-18 11:38:38 +08:00
lindsayshuo
c1b5cfdf2d
Update yolov8_cls_trt.py ( #1455 )
...
batch image data
2024-03-14 13:53:42 +08:00
wang-xinyu
5d7896913c
update readme
2024-03-12 18:42:23 +08:00
lindsayshuo
01191f5e61
yolov8 cls ( #1453 )
...
* yolov8 cls
* Update README.md
* Update yolov8_cls.cpp
* Update block.h
* Update block.cpp
* Update yolov8_cls.cpp
---------
Co-authored-by: Wang Xinyu <wangxinyu_es@163.com>
2024-03-12 17:53:07 +08:00
Wang Xinyu
04d3a931f5
Update README.md
2024-03-11 13:01:58 +08:00
WuxinrongY
e73bffcd25
tensorrt-yolov9 ( #1449 )
...
* tensorrt-yolov9
* format change
* change format
* Update block.cpp
* format: add space
* update block.cpp: add space
* update config.h
---------
Co-authored-by: Wang Xinyu <wangxinyu_es@163.com>
2024-03-11 12:58:47 +08:00
Wang Xinyu
e585b15c36
Update README.md
2024-03-07 19:06:42 +08:00
AadeIT
aa64535e1d
add csrnet ( #1450 )
...
* add csrnet
* Add update result jpg
CSRNet Inference result
* fix pr format
* add density plot code and update README.md
fix img src
---------
Co-authored-by: liulf <liulf@nncsys.com>
2024-03-07 19:00:37 +08:00
Wang Xinyu
eec1ff9e96
Update README.md
2024-03-06 18:40:29 +08:00
AMIYA MAITY
ec64a80b4b
Yolov7-Changed eps value ( #1444 )
...
* Changed eps value
* Update preprocess.h
---------
Co-authored-by: Wang Xinyu <shaywxy@gmail.com>
2024-03-06 18:38:59 +08:00
spacewalk01
4dae28be20
fix the problem of destroying a builder object ( #1443 )
2024-02-28 16:48:11 +08:00
Mehmet Onurcan KAYA
432a6729a7
YOLOv7 kNumAnchor correction ( #1435 )
...
* kNumAnchor correction
It was not working for number of anchors != 3
* kNumAnchor correction
2024-01-26 13:21:30 +08:00
wang-xinyu
5336af0551
yolov8: fix destroy order
2024-01-23 10:19:39 +08:00
Mehmet Onurcan KAYA
ad2c7a14d0
kNumAnchor correction ( #1429 )
...
It was not working for number of anchors != 3
2024-01-16 14:49:27 +08:00
Wang Xinyu
2df7dd7f91
Update postprocess.cu
2023-11-14 18:36:17 +08:00
Wang Xinyu
f39f2c1433
Update yolov8_det_trt.py
2023-11-03 16:18:36 +08:00
Wang Xinyu
06d3a40a78
Update README.md
2023-11-02 09:50:45 +08:00
lindsayshuo
1d3862d99c
Fix cuda post processing ( #1395 )
...
* Fix cuda post processing
Fix cuda post processing
* Update postprocess.cu
---------
Co-authored-by: Wang Xinyu <shaywxy@gmail.com>
2023-11-01 15:00:13 +08:00
lindsayshuo
2fafde1af6
Fix the issue of int8 quantization compilation errors in YOLOv8 ( #1393 )
...
* Fix the issue of int8 quantization compilation errors in YOLOv8
Fix the issue of int8 quantization compilation errors in YOLOv8
* Update model.cpp
* Update model.cpp
---------
Co-authored-by: Wang Xinyu <shaywxy@gmail.com>
2023-10-30 15:26:53 +08:00
emptysoal
120121891d
modify yolov8 cpu postprocess iou ( #1390 )
2023-10-30 15:15:21 +08:00
Rex
c997e35710
yolov8n ( #1387 )
...
* 修改yolov8nchannel数量以及 coco_file引用传递
* 优化代码风格
* 优化代码风格
2023-10-23 10:56:10 +08:00
Wang Xinyu
6a267e0413
Update README.md
2023-10-17 21:04:00 +08:00
Rex
9a53b7f226
Yolov8 seg ( #1381 )
...
* v8_seg
* fix code style
* fix code style
* [优化代码风格]
* readme中增加yolov8_seg分割的使用
* 更新readme
* 更新readme
* [fix python infer bug]
* Update README.md
* Update README.md
---------
Co-authored-by: Wang Xinyu <shaywxy@gmail.com>
2023-10-17 20:59:21 +08:00
Charles_Huan
7b95f981f6
Change resize to letterbox to keep model pre-processing consistent ( #1371 )
...
* Update model.cpp
Fix class mode pooling layer
* Change resize to letterbox to keep model pre-processing consistent
Change resize to letterbox to keep model pre-processing consistent
* Update yolov5_cls.cpp
reuse the preprocess function
* Update calibrator.cpp
* Update calibrator.h
* Update yolov5_cls.cpp
* Update yolov5_cls.cpp
* Update calibrator.h
* Update yolov5_cls.cpp
---------
Co-authored-by: Wang Xinyu <shaywxy@gmail.com>
2023-10-11 15:05:01 +08:00
Wang Xinyu
3271725dd4
Update README.md
2023-09-27 18:40:38 +08:00
Charles_Huan
67b5b7a9ea
Update model.cpp ( #1369 )
...
Fix class mode pooling layer
2023-09-27 18:37:55 +08:00
何恺悦
eef6c5a711
fix builder->destroy() error in TRT8.6 ( #1368 )
2023-09-27 18:30:17 +08:00
lindsayshuo
12c19acce8
retinafaceAntiCov tensorrt8 ( #1366 )
...
* retinafaceAntiCov tensorrt8
* Update decode.h
2023-09-27 18:29:18 +08:00
lindsayshuo
24811e0520
Fix network issues with yolov8n ( #1365 )
2023-09-22 17:31:20 +08:00
lindsayshuo
3cbb7cd257
Fix abnormal network structure issues in Yolov8n and add inference function for multiple batches ( #1338 )
...
* Fix abnormal network structure issues in Yolov8n and add inference function for multiple batches
* repair function
* repair function
2023-07-21 14:30:11 +08:00
lindsayshuo
ae3bd5e6b2
Move GPU post-processing separately to cuda_postprocess.cu ( #1331 )
...
* Move GPU post-processing separately to cuda_postprocess.cu
* Update postprocess.cpp
* Update postprocess.cu
---------
Co-authored-by: wang-xinyu <wangxinyu_es@163.com>
Co-authored-by: Wang Xinyu <shaywxy@gmail.com>
2023-07-18 13:23:56 +08:00
wang-xinyu
e9a972e332
add symlink
2023-07-14 17:04:08 +08:00
lindsayshuo
70d0adc9ec
Yolov8 adds cuda post-processing function ( #1326 )
...
* Yolov8 adds cuda post-processing function
* Yolov8 adds cuda post-processing function
* Yolov8 adds cuda post-processing function
* Yolov8 adds cuda post-processing function
* Yolov8 adds cuda post-processing function
* Yolov8 adds cuda post-processing function
* Yolov8 adds cuda post-processing function
* Yolov8 adds cuda post-processing function
* Yolov8 adds cuda post-processing function
* Update postprocess.cpp
* fix user id
* Yolov8 adds cuda post-processing function
* Yolov8 adds cuda post-processing function
* Yolov8 adds cuda post-processing function
* Yolov8 adds cuda post-processing function
* Yolov8 adds cuda post-processing function
* Delete images
* Update postprocess.h
* Update postprocess.cpp
---------
Co-authored-by: Wang Xinyu <shaywxy@gmail.com>
2023-07-13 13:13:56 +08:00
lindsayshuo
bc4f854dc3
Add an engine export that can customize category models ( #1322 )
...
* Add an engine export that can customize category models
* Add an engine export that can customize category models
2023-07-05 10:37:09 +08:00
Prince Patel
30ea76e395
Update README.md ( #1298 )
...
gd gw are not required in yolov7
2023-07-03 10:24:03 +08:00
Wang Xinyu
4c9ef43ba1
Update README.md
2023-07-03 10:11:40 +08:00
Wang Xinyu
be8f6dc66c
Update README.md
2023-06-30 15:03:23 +08:00
lindsayshuo
fee7cbd0fe
yolov8 ( #1321 )
...
* yolov8
* Update README.md
* Update README.md
* Update main.cpp
* Delete yolov8/output directory
* Delete yolov8/output directory
2023-06-30 15:01:55 +08:00
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
