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

146 Commits

Author SHA1 Message Date
Francesco Gatti df31375b67 Merge branch 'master' into tensorrt8 2022-03-30 16:41:45 +02:00
Francesco Gatti 69bb7370a5 compile with tensorrt7 2022-03-30 15:52:30 +02:00
perseusdg 480b5a9c5a added monodepth2_1024 2022-02-22 14:50:39 +00:00
Harshvardhan Chandirasekar 3e86671c50 Moved batchnorm and test_monodepth2_new_format layer to dev 
Signed-off-by: perseusdg <harshvardhan.chandira@gmail.com>
 2022-01-24 21:41:59 +05:30
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 19e41a8b99 monodepth2 new format (test) conv + independent batch norm 2022-01-11 12:38:36 +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 7298dcfb2f Added monodepth2.cpp 2022-01-06 16:30:11 +00:00
Micaela Verucchi 04de9908a6 Add yolo4-csp for crowdhuman dataset, add shelfnet for coco-stuff dataset, fix minor in demo.cpp 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2021-11-29 17:33:57 +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 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 c0e2097397 - CMakeLists.txt opencv cuda contrib autodetect 
- Updated Docker to cuda-11.3+cudnn-8.2.1+TensorRT-8.0.34,Ubuntu to 20.04 and OpenCV to 4.5.4
- Updated OpenCV4 to 4.5.4 in install_OpenCV4.sh
- Updated README.md
 2021-11-10 12:30:57 +05:30
perseusdg 744396fb0e Update README.md,windows.md and demo.cpp 
Small fixes in DeformableConvRT.cpp
 2021-11-09 13:37:26 +05:30
perseusdg 802c01bd8f windows debug fix 2021-10-30 19:29:21 +05:30
perseusdg 8c36dd0431 mobilenet works with trt7(IPluginV2IOExt) ,need to test it with trt8 2021-10-27 18:00:33 +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 eba78e7e78 Updates to build libkernel.so under TensorRT 8 2021-08-16 13:59:19 +05:30
Davide Sapienza 0292e21c33 Merge remote-tracking branch 'origin/master' into cnet 2021-07-22 17:03:23 +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 84ff978ccb Merge with master, all tests passed 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2021-07-20 12:48:41 +02:00
Francesco Gatti c306b36860 Update CMakeLists.txt 2021-06-17 15:31:25 +02:00
Francesco Gatti 9b78f143cb Update CMakeLists.txt 2021-06-16 14:53:46 +02:00
Francesco Gatti d847a4b852 Update CMakeLists.txt 2021-06-15 20:33:41 +02:00
Francesco Gatti 8df7d5fd1b Update CMakeLists.txt 2021-06-15 20:31:58 +02:00
Francesco Gatti 6d7c456f72 cmake version fix 2021-06-15 19:39:34 +02:00
Francesco Gatti b0fdeb4127 install targets 2021-06-15 19:38:13 +02:00
Francesco Gatti 6b8ae1e27c remove wrong cuda arch that cause big performance gap 
fix #226
 2021-05-16 22:16:21 +02:00
Davide Sapienza 34c1c3d577 Update cnet branch. 
This commit splits the demo3D in two demo: one for the 3D object
detection and one for the tracking.

It renames the files related to CenterTrack.

It adds a new parameter to select the tracker mode (2D or 3D).

Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2021-05-11 16:17:23 +02:00
Fabio Bagni f8327e2dac Merge with cnet 2021-05-04 11:50:08 +02:00
Davide Sapienza 28fab9c3e1 Merge remote-tracking branch 'origin/master' into cnet 2021-04-27 14:43:39 +02:00
Harshvardhan Chandirasekar f90ee8ab7d Update CMakeLists.txt 2021-04-09 13:20:44 +05:30
perseusdg 1de804f98d Code cleanup and readme fixes 2021-04-09 13:17:41 +05:30
hchandirasekar 44b71ae6f3 ReadMe.md windows changes 2021-03-25 20:44:43 +05:30
hchandirasekar 78859fe191 timer fix 2021-03-25 16:38:53 +05:30
Harshvardhan Chandirasekar 304ab49897 shared_ptr migrations 2021-03-17 22:26:26 +05:30
hchandirasekar 6aa8666be5 Commits for msvc 16.9 2021-03-10 12:59:04 +05:30
hchandirasekar 4a90314333 minor fixes in test.h 2021-01-26 23:11:58 +05:30
hchandirasekar 2d4dececb6 Builds on windows successfully,issues with deserialization and downloading weights 2021-01-26 20:17:43 +04:00
hchandirasekar fb52444cdc Replaced dynamic arrays with std::vector ,works on linux ..needs to be tested on windows after clearing up the lnk2019 error 2021-01-25 04:12:05 +05:30
perseusdg 56feb54377 able to build kernels as shared object file(dll),and minor changes to lstm.cpp and utils.cpp to overcome minor msvc build errors 2021-01-21 00:22:15 +04:00
Davide Sapienza 9f10c3f6e2 Add CenterTrack based on dla34. 
The implemented network refers to the nuScenes_3Dtracking.pth model.

Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-12-07 17:59:27 +01:00
Davide Sapienza 7f65ee0b2a Merge remote-tracking branch 'origin/master' into cnet 
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-12-07 17:45:44 +01:00
Micaela Verucchi a52e18b6e6 Add shelfnet_mapillary, README_seg, resize of input 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-11-24 12:37:33 +01:00
Micaela Verucchi a5cc4e3eda Add berkeley test, add weights for shelfnets 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-07-01 10:53:24 +02:00
Micaela Verucchi 082920f3f5 Shelfnet works, also visualization. Postprocessing need to be parallelized 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-06-23 20:01:47 +02:00