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

109 Commits

Author SHA1 Message Date
Harshvardhan Chandirasekar 6a133d8dec -Replaced ISliceLayer based paddings(reflection,constant and zero) with the IPluginV2 version for TensorRT >= 8.2.0 
-Fixed demo build issue on Windows

Signed-off-by: perseusdg <harshvardhan.chandira@gmail.com>
 2022-01-20 00:52:22 +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
Micaela Verucchi bbae618118 Update README and add print in demo 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2021-11-24 18:08:41 +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
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 c5e66c6bf6 TRT8 works with almost every nerual network now!!!!(including demo3d) 2021-10-28 23:35:37 +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
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 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
Micaela Verucchi 318dfe4b55 merge with master 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2021-07-20 18:38:46 +02:00
Micaela Verucchi 69256992a3 Add colormap for each segmentation dataset 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2021-07-20 15:29:26 +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
Micaela Verucchi b3c44a86a8 Fix shelfnet berkeley, update readme, minors 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2021-07-20 10:57:53 +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
Davide Sapienza 0dc96d2a9e Improve CenterTrack. 
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2021-05-04 17:59:20 +02:00
Davide Sapienza 2367519799 Add the calibration matrix reading for CenterTrack 
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2021-04-30 17:10:51 +02:00
Davide Sapienza be6ad27c11 Batch size > 1 for the 3D demo. 
This commit lets to use differtent batch size for 3D CenterNet
and CenterTrack.

Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2021-04-29 11:13:24 +02:00
Davide Sapienza 28fab9c3e1 Merge remote-tracking branch 'origin/master' into cnet 2021-04-27 14:43:39 +02:00
perseusdg 1de804f98d Code cleanup and readme fixes 2021-04-09 13:17:41 +05:30
perseusdg 37b2a5bd98 minor modifications 2021-04-08 00:57:58 +05:30
hchandirasekar 2e92944f1d Opencv cuda fix 2021-03-31 03:44:51 -07:00
hchandirasekar e94e1f7622 tkdnn first patch for windows 2021-03-24 22:26:46 +05:30
hchandirasekar 6aa8666be5 Commits for msvc 16.9 2021-03-10 12:59:04 +05:30
hchandirasekar 2d4dececb6 Builds on windows successfully,issues with deserialization and downloading weights 2021-01-26 20:17:43 +04:00
perseusdg 512acd8cba minor fixes 2021-01-21 09:29:27 +04:00
Davide Sapienza 1cfa199ee6 Add pre-processing and post-processing stats 
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-12-09 19:43:51 +01:00
Davide Sapienza 9e1d7b3bb4 Update demo3d with show flag 
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-12-09 18:05:54 +01:00
Davide Sapienza 48ecebe6dd Add CenterTrack pre, post, visualization and demo. 
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-12-07 18:45:29 +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 04b4a69107 Merge with master 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-09-14 17:30:02 +02:00
Micaela Verucchi f778e1aa99 Fixed boxes to float, add conf thresh as param 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-08-05 19:55:10 +02:00
Micaela Verucchi 286e777300 Improved segmentation results, removed resize, code to reorder 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-07-22 19:21:46 +02:00
Micaela Verucchi 594947f301 Update file names map demo 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-07-08 11:27:09 +02:00
Micaela Verucchi e7779ad773 Add conf_thresh parameter to detectors, add batches handling in map demo 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-07-08 09:52:36 +02:00
Micaela Verucchi b4c8c2bbad Add stats 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-07-03 18:45:20 +02:00
Micaela Verucchi 79cd96de6f Add resize to original size, writing of segmentation 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-07-02 12:14:20 +02: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 d25803d438 Post-processing improved 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-06-24 18:46:25 +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
Micaela Verucchi be9e327aef Fix minor, update readme 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-06-09 21:01:19 +02:00
Micaela Verucchi a0e4e9f209 Modify yolov4_berkeley link for download, clean to merge with master 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-06-01 15:17:53 +02:00
Davide Sapienza 6bdf47bae6 Add 3D demo program 
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-05-27 18:05:38 +02:00
Micaela Verucchi 377310af50 Add variable batch for detector update 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-05-19 09:32:27 +02:00
Micaela Verucchi 23b40de508 Add verbose define 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-05-15 09:41:27 +02:00