Commit Graph

447 Commits

Author SHA1 Message Date
Wang Xinyu
b0be525f9b
Update README.md 2024-07-03 16:36:49 +08:00
KRG
ed052a93dc
Update yolov8_seg.cpp for 'get_downscale_rect' to fix empty mask issue. (#1549)
Fix issue for https://github.com/wang-xinyu/tensorrtx/issues/1547
2024-07-03 09:57:24 +08:00
Wang Xinyu
fd2fab22f9
Update README.md 2024-06-21 15:47:58 +08:00
WuxinrongY
e065c7508d
yolov9 t s m (#1541)
* yolov9 t s m

* change chinese
2024-06-21 15:44:24 +08:00
lindsayshuo
cc725017a4
repaire python infer code (#1526)
* Add the generation of multi-class pose engines

* Change grids in forwardGpu to one-dimensional arrays

* Update README.md

* repaire python infer code

* Update yolov8_det_trt.py

* Update yolov8_pose_trt.py

* Update yolov8_seg_trt.py

* Optimize parameters

* Optimize parameters

* Optimize parameters

---------

Co-authored-by: lindsayshuo <lindsayshuo@foxmail.com.com>
2024-05-15 10:13:54 +08:00
lindsayshuo
5cb97ff017
Change grids in forwardGpu to one-dimensional arrays and fix p2 model download link (#1524)
* Add the generation of multi-class pose engines

* Change grids in forwardGpu to one-dimensional arrays

* Update README.md

---------

Co-authored-by: lindsayshuo <lindsayshuo@foxmail.com.com>
2024-05-14 13:48:38 +08:00
lindsayshuo
e555c9f93e
Solving the issue of out of bounds caused by detection boxes less than 0 when filling colors in segmentation masks (#1516)
* Add the generation of multi-class pose engines

* Solving the issue of out of bounds caused by detection boxes less than 0 when filling colors in segmentation masks

* Solving the issue of out of bounds caused by detection boxes less than 0 when filling colors in segmentation masks

---------

Co-authored-by: lindsayshuo <lindsayshuo@foxmail.com.com>
2024-05-09 16:35:14 +08:00
lindsayshuo
f201451f85
Modify get_rect_adapt_landmark to adaptively decode any number of key points abd repair gen_wts.py (#1514)
* Add the generation of multi-class pose engines

* Modify get_rect_adapt_landmark to adaptively decode any number of key points abd repair gen_wts.py

---------

Co-authored-by: lindsayshuo <lindsayshuo@foxmail.com.com>
2024-05-07 19:13:15 +08:00
lindsayshuo
f2e8b2bc62
Add the generation of multi-class pose engines (#1512)
Co-authored-by: lindsayshuo <lindsayshuo@foxmail.com.com>
2024-05-06 11:19:08 +08:00
lindsayshuo
37633b4529
pose int8 images path and fix pose p6 (#1507)
* pose int8 images path and fix pose p6

* pose int8 images path and fix pose p6
2024-04-29 14:30:27 +08:00
Wang Xinyu
ae38073988
Update README.md 2024-04-29 11:37:03 +08:00
ChangjunDAI
812dc0bb6e
add quantization folder path (#1503) 2024-04-29 11:30:28 +08:00
lindsayshuo
88f50ab7e6
yolov8_pose_trt.py (#1504)
* yolov8 p2

* yolov8 p2

* yolov8 p2

* yolov8 p2

* Update yolov8_det.cpp

* Update model.cpp

* Update model.cpp

* Update model.cpp

* Update model.cpp

* fix strides

* yolov8 pose

* yolov8 pose

* yolov8 pose

* yolov8 pose

* yolov8 pose

* yolov8 pose

* yolov8 pose

* yolov8 pose

* yolov8 pose

* yolov8 pose

* yolov8 pose

* yolov8 pose

* yolov8_pose_trt.py

* yolov8 pose p6
2024-04-29 11:23:29 +08:00
Wang Xinyu
e97add4757
Update README.md 2024-04-28 14:58:44 +08:00
lindsayshuo
af955ed886
yolov8 pose (#1502)
* yolov8 p2

* yolov8 p2

* yolov8 p2

* yolov8 p2

* Update yolov8_det.cpp

* Update model.cpp

* Update model.cpp

* Update model.cpp

* Update model.cpp

* fix strides

* yolov8 pose

* yolov8 pose

* yolov8 pose

* yolov8 pose

* yolov8 pose

* yolov8 pose

* yolov8 pose

* yolov8 pose

* yolov8 pose

* yolov8 pose

* yolov8 pose

* yolov8 pose
2024-04-28 14:55:26 +08:00
Wang Xinyu
47baebce60
Update README.md 2024-04-23 19:42:41 +08:00
WuxinrongY
bfe13faac7
set default FP16 and update GELAN (#1494)
* set default FP16 and update GELAN

* use pre-commit

* use pre-commit-V2
2024-04-23 18:39:10 +08:00
lindsayshuo
d033a63f07
fix strides (#1492)
* yolov8 p2

* yolov8 p2

* yolov8 p2

* yolov8 p2

* Update yolov8_det.cpp

* Update model.cpp

* Update model.cpp

* Update model.cpp

* Update model.cpp

* fix strides
2024-04-23 10:22:29 +08:00
Wang Xinyu
d1a184ffca
Update README.md 2024-04-22 13:42:25 +08:00
Wang Xinyu
b3814e13de
Update README.md 2024-04-22 13:41:03 +08:00
Wang Xinyu
d4aa52db68
Rename readme.md to README.md 2024-04-22 13:27:06 +08:00
wang-xinyu
739c5a4061 update readme 2024-04-22 13:24:16 +08:00
B1SH0PP
8bc47d8b06
add an anomaly detection network efficient_ad (#1490) 2024-04-22 11:36:12 +08:00
Wang Xinyu
417f6faf0f
Update README.md 2024-04-18 10:23:49 +08:00
lindsayshuo
20f46afd9d
yolov8 p2 (#1487)
* yolov8 p2

* yolov8 p2

* yolov8 p2

* yolov8 p2

* Update yolov8_det.cpp

* Update model.cpp

* Update model.cpp

* Update model.cpp

* Update model.cpp
2024-04-18 10:21:53 +08:00
Zhang-Wen-chao
c889b84df2
Fix issue 1405 for yolov7 (#1484) 2024-04-10 18:26:05 +08:00
lindsayshuo
62050de3e7
yolov8 detection p6 (#1479)
* yolov8 detection p6

* Update README.md

* update README.md

* Update README.md

* Update README.md

* Update README.md

* Update README.md

* update README.md
2024-04-01 10:39:14 +08:00
Wang Xinyu
c86d808e52
add python flake8 check (#1470)
* fix code style

* add python flake8 check

* add python flake8 check
2024-03-27 13:22:16 +08:00
stbidtf
64297a268e
Fixed a bug with inconsistent inference results (#1468) 2024-03-26 19:38:22 +08:00
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