Commit Graph

430 Commits

Author SHA1 Message Date
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
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