mmr/tkDNN
Archived
1
0
Fork 0
Code Issues 46 Pull Requests 12 Releases 4 Activity
608 Commits 11 Branches 5 Tags
56da10df647bd5cc7984a0b16520e25c2e87dfe2
Commit Graph

306 Commits

Author SHA1 Message Date
Micaela Verucchi 56da10df64 (tkDNN): Support Resnet-101-AP-GeM for CUDNN, TRT to be fixed 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2022-09-24 16:57:12 +02:00
Francesco Gatti d4f7b4ad8b remove using namespace useless 2022-03-30 22:14:06 +02:00
Francesco Gatti 5e71b99265 YoloRT save bias, mask and clasesName into RT file 2022-03-30 20:46:51 +02:00
Francesco Gatti fa9db167b8 version 0.7 2022-03-30 17:48:02 +02:00
Francesco Gatti df31375b67 Merge branch 'master' into tensorrt8 2022-03-30 16:41:45 +02:00
Harshvardhan Chandirasekar 40266a6c32 Fixed tensorrt8 branch to work jetpack 4.5 and tensorrt7 
Signed-off-by: perseusdg <f20180523@goa.bits-pilani.ac.in>
 2022-03-16 18:09:32 +05:30
perseusdg b3bc93693f Revert "Merge branch 'tensorrt8-rds' into depth" 
This reverts commit 00f06f7bcc, reversing
changes made to 6a133d8dec.
 2022-01-25 18:30:39 +05:30
Harshvardhan Chandirasekar afdad8e661 Moved batchnorm and test_monodepth2_new_format layer to dev 
Signed-off-by: perseusdg <harshvardhan.chandira@gmail.com>
 2022-01-24 21:48:21 +05:30
Harshvardhan Chandirasekar 00f06f7bcc Merge branch 'tensorrt8-rds' into depth 2022-01-20 22:05:54 +05:30
Harshvardhan Chandirasekar 53725c88a9 Merge pull request #3 from perseusdg/tensorrt8 
push tensorrt8 commits to the rds branch
 2022-01-20 21:12:49 +05:30
Micaela Verucchi decd73d298 Fix saving result video in demoDepth, add automatic download for monodepth2 weights 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2022-01-19 10:22:06 -08:00
Micaela Verucchi 907df27e07 Fix monodepth2, add demoDepth: 
- Fix monodepth2 network, now works with both cuDNN and tensorRT
- Substitute cuDNN ELU with tkDNN one
- add DepthNN class
- add demoDepth demo, now only works with monodepth2 net

Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
               Francesco Gatti <gattifrancesco@hotmail.it>
 2022-01-18 21:34:16 -08:00
perseusdg 061bc79a69 Added BatchNorm to NetworkRT (conver_layer) 2022-01-10 22:48:16 +05:30
perseusdg bcf0c4eab3 Added BatchNorm Layer (Testing still needs to be done) 2022-01-10 22:48:06 +05:30
perseusdg e1eac2d42a Added a seperate layer file for batchnorm,to support an independent batchnorm class in order to pass necessary parameters read from the bin file. 2022-01-10 22:47:59 +05:30
perseusdg 3023926695 fix constantpadding commit 2022-01-09 22:17:43 +05:30
perseusdg 4d8f99b441 depth->tensorrt8 patches 2022-01-09 19:24:38 +05:30
perseusdg 3b58fbcb82 Added ConstantPadding plugin for TensorRT < 8.2 2022-01-07 13:05:09 +05:30
perseusdg f189efcbb1 Reflection Padding native plugin fix ,forgot to added input_dim.c in the plugin creator 2022-01-07 05:41:26 +00:00
perseusdg b75cecb105 Added constant and zero padding with ISliceLayer,need to add them for tensorrt versions less than 8.2 using IPluginV2Ext instead of ISliceLayer since they dont seem to support reflect and zero 2022-01-04 09:03:04 +05:30
perseusdg ba022663f1 completed padding migrations from github 2022-01-03 19:49:32 +05:30
perseusdg 6837644eb2 add reflection padding from github 2022-01-03 18:58:13 +05:30
perseusdg dcf4054bc6 performance improvements for yolo based networks,significant reduction in inference time can be seen yolo4tiny ,yolo4 and minor reduction in inference time can be seen in yolo4_berkeley_f1 and yolo4_berkeley - tested with a batchsize of 1 and 2 and on gtx 1070,it is possible that the performance improvement is more signficant in newer hardware 2022-01-03 13:52:16 +05:30
Micaela Verucchi a4781244f4 Fix merge problem 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2021-11-24 18:16:15 +01:00
Micaela Verucchi 0707c26bbd Merge of perseusdg-tensorrt8 inside tkDNN 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2021-11-24 17:59:27 +01:00
perseusdg 9cecc5051a -Modified Dockerfile.base to cudagl 
-Changed demo to take in input from demoConfig.yaml file
-Readme changes for demo.md
 2021-11-24 03:39:57 +05:30
Micaela Verucchi be5864748a Use yaml config file for the demo instead of param list 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2021-11-23 16:29:40 +01:00
Micaela Verucchi 55df97afe1 Fix warnings, upgrade to C++14 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2021-11-23 13:01:07 +01:00
perseusdg a8c98e3c31 Signed-off-by: perseusdg <43143075+perseusdg@users.noreply.github.com> 
removed all TRT8_DEPRACTED functions
 2021-11-19 22:40:52 +05:30
perseusdg 19f12d9c15 rds_slam -> tensorrt8 port for rds 2021-11-15 00:08:37 +05:30
perseusdg 6f936096ae Added num of layers in net and netRT,suggestions from pull request 270 on main repo 2021-11-14 07:13:42 +05:30
perseusdg ee5000ccca bug fixes for dla networks and ported optimization from different pull request 2021-11-13 02:00:49 +05:30
perseusdg 9e328c0daa Migrate tkDNN max pooling plugin creation to pluginRegistry from the default method 2021-11-10 20:09:57 +05:30
perseusdg c5e66c6bf6 TRT8 works with almost every nerual network now!!!!(including demo3d) 2021-10-28 23:35:37 +05:30
perseusdg 8c36dd0431 mobilenet works with trt7(IPluginV2IOExt) ,need to test it with trt8 2021-10-27 18:00:33 +05:30
perseusdg 54e7af11ed tensorrt7 support for ipluginv2 2021-10-18 18:07:57 +05:30
perseusdg 18cc6abbb6 DeformableConvRT.h fixes ,cmake cuda arch auto detection 2021-09-07 23:57:52 +05:30
perseusdg 03473743c4 tkDNN works with trt8!!,need to test int8 and mobilenet,dla_cnet (fps seems to be a bit low 350 on trt8 compared to 396 on trt7) 2021-09-07 22:00:28 +05:30
Harshvardhan Chandirasekar 65ba5c9844 tkDNN can now deserialize tensorrt-8 engine (both through test_* and trtexec) 
but demo has issues in yolo::computeDetections
 2021-09-05 01:06:56 +05:30
perseusdg de83ae5d25 update tensorrt8 branch 2021-08-30 19:04:26 +05:30
perseusdg 2ffe07057e Mnist works at the moment with trt8,others like yolo4tiny and mobilenet generate the engine files but crash after throwing nvifer1::CudaRuntimeError and when demo is being run ,it doesnt deserialize properly and crashes 2021-08-29 03:18:58 +05:30
perseusdg 3d8b1ac494 Yolo3Detection.cpp doesn't build yet,issues with dependecies of preprocessing and postprocessing on IPluginFactory 2021-08-21 22:29:36 +05:30
perseusdg eba78e7e78 Updates to build libkernel.so under TensorRT 8 2021-08-16 13:59:19 +05:30
perseusdg d2ce966c46 fix return type for bool functions 2021-08-04 12:37:58 +05:30
perseusdg 1412aa66c4 minor fixes for windows ,for release v0.6 2021-07-24 08:51:44 -07:00
Davide Sapienza 0292e21c33 Merge remote-tracking branch 'origin/master' into cnet 2021-07-22 17:03:23 +02:00
Davide Sapienza b9f82510d3 Add demo_utils. 
This commit splits the utils file into two different files:
utils and demo_utils, to overcome some opencv problems caused
by demo_utils includes.

Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2021-07-22 10:45:18 +02:00
Davide Sapienza 1216e8bb74 Add camera calibration file reading for demo3d and demoTracker. 
Some minor fixes.

Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2021-07-21 16:14:53 +02:00
Davide Sapienza be2d361ac6 Update 3d detection 
This commit updates the CenternetDetection3d class and fixes
some bugs.

It also removes the resnet101 network for CenterNet 3D.
This network doesn't exist.

Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2021-07-21 15:51:13 +02:00
Micaela Verucchi 318dfe4b55 merge with master 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2021-07-20 18:38:46 +02:00