Wang Xinyu
3a40bbc184
Update README.md
2024-10-22 11:44:42 +08:00
Wang Xinyu
3f76507a9f
Update README.md
2024-10-22 11:44:25 +08:00
lindsayshuo
21f23ffcea
Distinguish the numclass of pose and obb in config. h and add the Oriented Bounding Boxes (OBB) Estimation algorithm ( #1593 )
...
* Add the generation of multi-class pose engines
* Change grids in forwardGpu to one-dimensional arrays
* Update README.md
* Update types.h
keypoints array with dynamic size based on kNumberOfPoints
* yolov8_5u_det(YOLOv5u with the anchor-free, objectness-free split head structure based on YOLOv8 features) model
* update
* fix code style
* yolov8_5u_det model download link
* yolov8_5u_det model download link
* Distinguish the numclass of pose and obb in config. h and add the Oriented Bounding Boxes (OBB) Estimation algorithm
* fix code style
---------
Co-authored-by: lindsayshuo <lindsayshuo@foxmail.com.com>
2024-10-22 11:37:34 +08:00
felipemasuda
0f75fd75f8
Update of instructions for using custom models in Yolov8 [README.md] ( #1590 )
...
* Update of instructions for using custom models in Yolov8 [README.md]
* Update README.md
Fix ultralytics<=8.2.103
2024-10-21 12:48:45 +08:00
mpj1234
4ce6f20693
add gen_wts ( #1591 )
2024-10-18 15:52:21 +08:00
Wang Xinyu
00e5b93c9d
Update README.md
2024-10-18 11:42:43 +08:00
zgjja
2050e3a061
[Refactor] Improve docker image build process and add some details ( #1586 )
...
* [Refactor] Improve docker image build process and add some details for pre-commit
* [Fix] Simple fixes from PR review
2024-10-18 10:57:55 +08:00
Wang Xinyu
79b02cbd24
Update readme.md
2024-10-11 17:02:50 +08:00
Wang Xinyu
563a5ff8ff
Update readme.md
2024-10-11 16:35:08 +08:00
Wang Xinyu
0676a97bed
Update README.md
2024-10-11 16:31:59 +08:00
mpj1234
4f79443dd2
Add the support of YOLO11' s det/cls/seg/pose in TensorRT8. ( #1584 )
...
* Add the support of YOLO11' s det/cls/seg/pose in TensorRT8.
* add train code link
2024-10-11 16:25:07 +08:00
Wang Xinyu
ac29dbb814
Update README.md
2024-10-09 18:51:19 +08:00
Phoenix
b671024a27
Add Ghostnet && Fix object destruction order in APIToModel function to avoid undefined behavior ( #1581 )
...
* add ghostnet
* add ghostnet
* add ghostnetV1&ghostnetV2
* Fix object destruction order in APIToModel function to avoid undefined behavior
* Fix pre-commit errors in ghostnet/README.md
* Fix pre-commit errors
* Fix pre-commit errors in mobilenetV3
* Add a noqa marker in mobilenet py files
2024-10-09 18:48:03 +08:00
lindsayshuo
dddc379143
yolov8_5u_det(YOLOv5u with the anchor-free, objectness-free split head structure based on YOLOv8 features) model ( #1577 )
...
* Add the generation of multi-class pose engines
* Change grids in forwardGpu to one-dimensional arrays
* Update README.md
* Update types.h
keypoints array with dynamic size based on kNumberOfPoints
* yolov8_5u_det(YOLOv5u with the anchor-free, objectness-free split head structure based on YOLOv8 features) model
* update
* fix code style
* yolov8_5u_det model download link
* yolov8_5u_det model download link
---------
Co-authored-by: lindsayshuo <lindsayshuo@foxmail.com.com>
2024-09-23 13:19:36 +08:00
lindsayshuo
e2545131df
keypoints array with dynamic size based on kNumberOfPoints ( #1570 )
...
* Add the generation of multi-class pose engines
* Change grids in forwardGpu to one-dimensional arrays
* Update README.md
* Update types.h
keypoints array with dynamic size based on kNumberOfPoints
---------
Co-authored-by: lindsayshuo <lindsayshuo@foxmail.com.com>
2024-08-28 17:28:13 +08:00
Wang Xinyu
d5230129c8
Update README.md
2024-08-21 15:03:30 +08:00
Lemonon
56ea9d944a
add new real-esrgan model real-esrgan-general-x4v3 ( #1566 )
...
* add real-esrgan-general-x4v3 for new real-esrgan model real-esrgan-general-x4v3
* Reorganize the files, change tensorrtx/real-esrgan/ and tensorrtx/real-esrgan-general-x4v3/ to tensorrtx/real-esrgan/x4plus/ and tensorrtx/real-esrgan/general-x4v3/.
---------
Co-authored-by: xuejiejie <xuejiejie@xiaoyingai.com>
2024-08-21 11:38:52 +08:00
Wang Xinyu
a6ef9e3da4
Update README.md
2024-07-29 16:36:10 +08:00
mpj1234
17d61f4b6e
TensorRT8 support for YOLOv10 ( #1561 )
...
* TensorRT8 support for YOLOv10
* delelet images, modify add test images in readme
2024-07-29 16:16:09 +08:00
Wang Xinyu
b4203e377e
Update pre-commit.yml
2024-07-11 12:59:47 +08:00
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
