BaofengZan
55ad6259ce
Fix compilation problems on Windows ( #644 )
2021-07-30 10:57:29 +08:00
JumpPandaer
abcc753b19
add detr-res101 support ( #645 )
...
* add detr
* Update README.md
* add int8 quantization
fix some known bugs
* add detr-res101 support
update detr README.md
update rcnn README.md
2021-07-26 00:37:42 +08:00
Jow Xiong
f81d5ba7f4
improperly use about stl at line 35 ( #621 )
2021-07-22 10:51:16 +08:00
Huang ZhuangZhuang
2e53f0cae7
update infer preprocessing ( #623 )
...
There are some problems with the preprocessing on LPRnet when infer, which will cause the predicted output results to be inconsistent with the pytorch version; as shown, the preprocessing has been updated here
2021-07-15 12:46:18 +08:00
Aymeric Dujardin
b7a754e2de
Fix for TensorRT 8 support ( #540 )
2021-07-05 12:02:56 +08:00
JumpPandaer
9bb2f6771c
add int8 quantization ( #612 )
...
* add detr
* Update README.md
* add int8 quantization
fix some known bugs
2021-07-05 11:15:36 +08:00
wang-xinyu
14e220f421
fix yolov5 get_depth round bug
2021-07-01 11:57:06 +00:00
Wang Xinyu
917ad809ae
Update README.md
2021-07-01 13:14:36 +08:00
JumpPandaer
161e6b05ee
add detr ( #606 )
...
* add detr
* Update README.md
2021-07-01 13:09:51 +08:00
JumpPandaer
7b85cfd392
fix bugs ( #602 )
...
see note in README
2021-06-28 10:31:59 +08:00
wang-xinyu
101e60348e
fix ofstream binary issue for windows #585
2021-06-16 07:58:13 +00:00
Wang Xinyu
f8c537586a
Update README.md
2021-06-11 10:42:16 +08:00
weiwei zhou
93e728769d
Add EfficientNet ( #590 )
...
* 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
* fix typo
* add hrnetseg w18 w32 w48
* add hrnetseg with ocr w18 w32 w48
* merge hrnet and small, add hrnet_ocr
* fix warning
* change project name
* add efficientnet
2021-06-11 10:34:40 +08:00
makaveli
668d89bbd5
fix: yolov5 get width ( #566 )
2021-05-25 20:47:16 +08:00
ybw108
d274d90722
Fix a bug of arcface-mobilefacenet ( #562 )
...
Fix a bug which can cause different output results from tensorrt and insightface(mxnet) arcface-mobilefacenet. Change the `eps` value of `BatchNorm2d` layer in `conv_bn_relu()`, `conv_bn()` and `DepthWise()` from `2e-5` to `1e-3` according to the origin mxnet implementation of mobilefacenet from insightface.
2021-05-24 18:31:41 +08:00
wang-xinyu
31646b6690
update readme
2021-05-24 03:49:13 +00:00
SsisyphusTao
d9e2cbc294
add centernet dla34 ctdet task. ( #558 )
...
* add centernet dla34 ctdet task.
* update readme.
* update readme and fix a bug in sample.
Co-authored-by: chandler <chandler@invix.com>
2021-05-23 16:29:14 +08:00
irvingzhang0512
1457e6fcb7
Fix a bug of arcface-mobilefacenet output size ( #556 )
2021-05-21 10:55:22 +08:00
wang-xinyu
4866607082
update readme
2021-05-17 11:44:48 +00:00
ybw108
53ba7e97b1
Add: arcface-mobilefacenet ( #545 )
...
* add arcface-mobilefacenet
* Update arcface-mobilefacenet.cpp
* modify CMmakeLists.txt, add arcface-r100.cpp
* fix typos,resolve conflicts,update README.md
Co-authored-by: Admin <huangz11@chinatelecom.cn>
2021-05-17 18:56:41 +08:00
Huang ZhuangZhuang
20fce2bce4
Update retina_r50.cpp and retina_nmt.cpp ( #546 )
2021-05-16 11:31:33 +08:00
Huang ZhuangZhuang
6a74b88b4e
update retinaface_trt del cvtColor and increase THRESH ( #535 )
...
* update retinaface_trt del cvtColor and increase THRESH
* update macro in retinafce_xx.cpp and preprocess_image description
* retinaface_xx.cpp add a space after
* HRESH for CONF and THRESHOLD for IOU is unified --THRESH
Co-authored-by: Admin <huangz11@chinatelecom.cn>
2021-05-11 19:57:11 +08:00
JumpPandaer
2c4fdea67d
update rcnn ( #521 )
...
* add MaskRcnn(C4)
* add MaskRcnnInference plugin for mask selecting
* split ROIHeads to BOXHead and MaskHead
* remove unuseful parameters in createEngine_rcnn and BuildRcnnModel
* change the type of scores_h, boxes_h and classes_h from unique_ptr to vector
* add doInference
* add maskrcnn postprocess
* update README.md
* update rcnn
* fix bugs with R18 and R34
add BasicBlock for R18 and R34
add STRIDE_IN_1X1, MakeStage is same with detectron2 now.
* update README.md
* update rcnn
replace picture with url
2021-05-07 10:50:55 +08:00
makaveli
f4c384d1b1
add: Scaled yolov4 ( #524 )
...
* add: mish, yololayer, layers
* update: CMake
* fix: compile
* add: yolov4-csp net def
* update: cuda kernel for scaled_yolov4
* increase nms thresh
* update: README
* add: gen_wts
* update: CMake
* fix memory leak
* Update README.md
* Update README.md
* Update README.md
* Update README.md
2021-05-06 11:05:20 +08:00
Aditya Lohia
4ffc56a99b
Add: Wideresnet ( #4 ) ( #518 )
...
* Adding WideResnet C++ (#2 )
* initialize wideresnet50
* add: wideresnet50 c++ code
* add: wide resnet python (#3 )
* add: wide resnet python
* fix: typo
Co-authored-by: makaveli <39617050+makaveli10@users.noreply.github.com>
Co-authored-by: makaveli <39617050+makaveli10@users.noreply.github.com>
2021-04-30 10:30:40 +08:00
weiwei zhou
1f7672ce5a
add hrnet semantic segmentation model w18, w32 and w48 ( #503 )
...
* 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
* fix typo
* add hrnetseg w18 w32 w48
* add hrnetseg with ocr w18 w32 w48
* merge hrnet and small, add hrnet_ocr
* fix warning
* change project name
2021-04-29 20:26:15 +08:00
wwqgtxx
b58a7987bc
add a simple warm_up in yolov5_trt.py ( #513 )
...
* add a simple warm_up in yolov5_trt.py
* remove if-else condition in infer()
* be careful with bgr and rgb image
2021-04-28 17:37:38 +08:00
Aditya Lohia
e8653a776d
Add: AlexNet, MobileNetV3, DenseNet121 Python Network Definition API ( #512 )
...
* add: mobilenetv2 Python network definition API
* add: mobilenetv3 base code
* add: mobilenetv2 Python network definition API
* restructure: mobilenetv2 code
* add: Alexnet Python Network Definition API
* update: README according to new folder architecture
* add: mobilenetv3 small and large python network definition API
* add: DenseNet121 Python Network Definition API
2021-04-28 13:39:55 +08:00
wang-xinyu
acc8a26213
yolov5 fix readme
2021-04-28 05:34:13 +00:00
wang-xinyu
117a16e291
yolov5 .py batch processing
2021-04-27 13:00:05 +00:00
wang-xinyu
9b60aa240f
Merge branch 'master' of https://github.com/wang-xinyu/tensorrtx
2021-04-27 08:39:48 +00:00
wang-xinyu
cb2aa5e72e
yolov5 .py get w and h from engine
2021-04-27 08:39:14 +00:00
makaveli
64c4b43f2f
fix: typo ( #509 )
2021-04-27 11:05:55 +08:00
wang-xinyu
c9dfbf2f7c
update readme
2021-04-26 12:29:44 +00:00
wang-xinyu
f088ec77d2
fix typo, add latency log
2021-04-26 11:38:57 +00:00
Aditya Lohia
bf1d851b5e
add: mobilenetv2 Python network definition API ( #506 )
...
* add: mobilenetv2 Python network definition API
* restructure: mobilenetv2 code
2021-04-26 14:04:22 +08:00
wang-xinyu
ad84d77335
update readme
2021-04-26 03:48:49 +00:00
wang-xinyu
9ff37c46ad
Update directory structure
2021-04-26 03:17:35 +00:00
makaveli
4d630ed599
Inceptionv4 ( #505 )
...
* add: structure inception v4
* fix: camelcase naming
* fix and optimize
* add: doInference def
* define inception layers
* fix: spacing
* add: inception network def
* fix layers
* create net
* validate
* add README
* update readme
* typo
* create dir inception
2021-04-26 11:06:52 +08:00
wang-xinyu
046dd5b417
yolov5 support s6/m6/l6/x6 models
2021-04-25 12:17:15 +00:00
wang-xinyu
3e9a4fc19b
verify yolov5 s/m/l/x and p6 backbone
2021-04-23 12:40:59 +00:00
wang-xinyu
e63a6a874f
small fix of yolov5
2021-04-23 08:34:10 +00:00
irvingzhang0512
501160d262
Support TSM-R50 C++ API ( #500 )
...
* C++ API for TSM-R50
* fix bugs
2021-04-23 14:57:01 +08:00
JumpPandaer
8d8e2c9b0c
add MaskRcnn(C4) ( #498 )
...
* add MaskRcnnInference plugin for mask selecting
* split ROIHeads to BOXHead and MaskHead
* remove unuseful parameters in createEngine_rcnn and BuildRcnnModel
* change the type of scores_h, boxes_h and classes_h from unique_ptr to vector
* add doInference
* add maskrcnn postprocess
* update README.md
2021-04-23 10:42:22 +08:00
JumpPandaer
5d6b09cf5a
update rcnn ( #495 )
...
add pixel_std in preprocess
fix synchronization error in roialign and cudaMemcpyAsync
improve coding style, limit line length less than 120 and so on
update README.md
upgrade TensorRT to 7.2
2021-04-21 10:55:08 +08:00
irvingzhang0512
8cfc8ee03f
Support TSM-R50 Python API ( #488 )
...
* add tensorrt temporal shift module and related pytorch implementations
* add .gitignore and getn weights script.
* rename get_wts.py script
* Add tsm-r50 demo.
* update readme
* remove useless codes
* update readme
* update readme
* remote video and .gitignore, update tutorial
* update readme and tutorial
* fix a few bugs and test on tensorrt 5.1
* update readme
2021-04-18 16:30:25 +08:00
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
