Commit Graph

222 Commits

Author SHA1 Message Date
Wang Xinyu
d9bdd7e59f
Update README.md 2021-04-16 11:47:16 +08:00
JumpPandaer
891202d458
add FasterRcnn(R50C4) (#475)
* Update yolov5.cpp

implement upsample with IResizeLayer

* Update yolov5.cpp

repair mistake and test it

* add FasterRcnn(R50C4)

maskrcnn is working

* add FasterRcnn(R50C4)

adjust indent

* update README

adjust coding style
2021-04-16 11:41:18 +08:00
Pavel Mikhadzionak
e72d9db48b
Fix synchronization error (#485) 2021-04-16 10:37:10 +08:00
irvingzhang0512
7f17ce4f9b
Build ResNet50 by TensorRT Python Network Definition API (#482)
* add tensorrt python api sample for resnet50

* update docs

* typo

* fix typo

Co-authored-by: Wang Xinyu <shaywxy@gmail.com>
2021-04-15 13:17:41 +08:00
irvingzhang0512
3c5939cd0d
add tensorrt python api sample for lenet (#476) 2021-04-14 15:34:57 +08:00
irvingzhang0512
f54f113fe8
typo (#474) 2021-04-13 18:21:41 +08:00
PeterH0323
12f7c1b8b1
Improve DBNet README.md (#467)
* Fix: syntax error in common.hpp , cuda & tensorrt not included in CMakeLists.txt

* Fix: layer warining while building in common.hpp and denet.cpp

* Fix: NOT droping mini boxes and low score boxes.

* Fix: The prediction boxes are NOT expanded by the specified proportion, which will lead to serious errors

* Improve DBNet README

* Improve DBNet README

* Improve DBNet README
2021-04-08 10:52:33 +08:00
JumpPandaer
66d641814e
Update yolov5.cpp (#463)
* Update yolov5.cpp

implement upsample with IResizeLayer

* Update yolov5.cpp

repair mistake and test it
2021-04-07 18:08:25 +08:00
PeterH0323
b74db15d87
Optimize dbnet (#464)
* Fix: syntax error in common.hpp , cuda & tensorrt not included in CMakeLists.txt

* Fix: layer warining while building in common.hpp and denet.cpp

* Fix: NOT droping mini boxes and low score boxes.

* Fix: The prediction boxes are NOT expanded by the specified proportion, which will lead to serious errors
2021-04-07 17:54:27 +08:00
wang-xinyu
5b90354d28 fix genwts file close 2021-04-07 05:21:37 +00:00
wang-xinyu
4070131934 update readme 2021-04-02 09:44:28 +08:00
Vineet
5d72c5af87
Add DenseNet121 (#460)
* add: densenet

* update readme
2021-04-01 20:21:29 +08:00
mingyu6yang
e381ae5d20
Retinaface python (#458)
* Create retinaface_trt.py

An example that uses TensorRT's Python api to make inferences.

* Update retinaface_trt.py

* Update README.md
2021-03-31 15:06:12 +08:00
Wang Xinyu
a7f43a304d
Update README.md 2021-03-17 19:29:45 +08:00
wuzuowuyou
4b2ef40380
tensorRT RefineDet (#443) 2021-03-17 11:19:22 +08:00
wang-xinyu
1336337589 also use google drive for model zoo and calib data 2021-03-08 18:44:50 +08:00
Wang Xinyu
eb1accc869
Create stale.yml 2021-03-05 18:49:04 +08:00
Wang Xinyu
7bf3545678 Update issue templates 2021-03-05 17:19:00 +08:00
wang-xinyu
2ae0fd1882 update readme 2021-03-05 15:10:42 +08:00
alpha
2c2d3f28ad
add LPRNet (#430)
* upload LRPnet

* upload LPRnet

* upload LPRnet

* 1
2021-03-04 17:29:19 +08:00
Wang Xinyu
577ca20128
Update README.md 2021-03-02 17:35:20 +08:00
Wang Xinyu
6b2a139af7
Update README.md 2021-03-02 17:06:17 +08:00
wang-xinyu
8e4de7dced optimize yolov5 cmdline args 2021-02-28 14:30:15 +08:00
AsakusaRinne
fec5e618d4
Another way to build yolov5 engines (#416)
* Update yolov5.cpp

* Update README.md

* Update README.md
2021-02-28 11:25:36 +08:00
BaofengZan
37f03a77c5
fixed some issues (#400)
* 1 动态输入
2 图像预处理和后处理和pytorch同步
3 代码格式统一

* 重新梳理代码

* tab 替换为4 space

* HRNet-Semantic-Segmentation

* update result img

* update readme

* Synchronize

* Modify some bugs
2021-02-07 17:41:16 +08:00
weiwei zhou
b15a45da39
Fix typo which causes "addConvRule not declared" (#387)
* fix typo
2021-01-31 17:26:33 +08:00
wang-xinyu
32d960a47c update readme 2021-01-31 16:37:09 +08:00
weiwei zhou
95bea3e7fa
add RepVGG, support all RepVGG predefined structures. (#384)
* create psenet

create psenet with weight from tensorflow

* delete some useless code

* repalce tab with 4 blanks

* fix network bug, rewrite post-processing pse algorithm

* update readme

* update readme

* add RepVGG
2021-01-31 10:57:30 +08:00
Victor Sonck
7fe5e135aa
Fix dockerfile bionic opencv 3.4 version (#383)
* Fix dockerfile bionic opencv 3.4 version

* add libraries needed for yolov5

* add libshape debug package

Co-authored-by: Victor Sonck <victor.sonck@ml6.eu>
2021-01-31 10:52:14 +08:00
wang-xinyu
1a82822474 fix and optimize 2021-01-29 16:20:48 +08:00
YuzhouPeng
6161f28854
add unet code (#380)
* add code

* remove txt
2021-01-28 12:59:02 +08:00
Ryo Mizote
4ae3e8743f
Fix a bug(wrong CMakeLists syntax) (#377) 2021-01-28 12:58:42 +08:00
wang-xinyu
5332e0e46c fix 2021-01-24 14:21:52 +08:00
wang-xinyu
c4fd9ca697 update readme 2021-01-24 14:12:50 +08:00
wang-xinyu
95b9b3bc9e yolov5 v4.0 int8 2021-01-24 13:58:06 +08:00
Wang Xinyu
86b8f5f701
update psenet readme 2021-01-22 19:47:38 +08:00
weiwei zhou
285ec22e5f
fix network bug and rewrite post-processing pse algorithm (#367)
* create psenet

create psenet with weight from tensorflow

* delete some useless code

* repalce tab with 4 blanks

* fix network bug, rewrite post-processing pse algorithm

* update readme

* update readme
2021-01-22 19:44:29 +08:00
Darren
556665cca1
IBN-Net (#362)
* IBN-Net

InstanceNorm2d

resnet50-ibna

resnet50-ibnb

* add ibnnet pytorch repo
2021-01-19 22:06:25 +08:00
wang-xinyu
7ba93e2d0d update readme, fps of yolov5 v4.0 2021-01-08 17:07:53 +08:00
wang-xinyu
d0a32e5646 yolov5 updated to v4.0 2021-01-08 16:52:48 +08:00
TeeVenDick
6d0f5cbf47
Fix typo (#350) 2021-01-07 20:54:56 +08:00
wang-xinyu
9c40d9570d rename files, update readme 2020-12-28 10:08:16 +08:00
BaofengZan
bc755aed10
HRNet-Semantic-Segmentation (#338)
add HRNet-Semantic-Segmentation
2020-12-27 10:28:56 +08:00
wwqgtxx
53ac268209
fix yolov3 build error (#335) 2020-12-21 11:27:57 +08:00
priteshgohil
a8e8391801
update yolov5/README.md (#333) 2020-12-21 11:27:28 +08:00
wang-xinyu
17efe30525 update readme 2020-12-17 16:07:44 +08:00
wang-xinyu
8565237890 fix cmake 2020-12-16 12:54:00 +08:00
LabMemNo003
a7f96eca6c
Fix typo in CMakeLists.txt (#329)
* Fix typo in CMakeLists.txt

* Fix format error for CUDA_NVCC_FLAGS
2020-12-14 10:32:48 +08:00
wang-xinyu
0b734e4818 update readme 2020-12-04 22:48:32 +08:00
weiwei zhou
cd277360e2
create psenet project with weight from tensorflow (#321)
* create psenet

create psenet with weight from tensorflow

* delete some useless code

* repalce tab with 4 blanks
2020-12-04 22:32:27 +08:00