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

100 Commits

Author SHA1 Message Date
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
Micaela Verucchi 40456592fc Adapt detection classes to use batches, adapt demos, update README 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-05-14 16:41:51 +02:00
Micaela Verucchi 2e3cb52cff Add flag to disable visualization and save video 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-05-11 15:08:10 +02:00
Micaela Verucchi 533bb48789 Add json detection creation for codalab check 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-05-11 11:57:58 +02:00
Francesco Gatti 85588f7343 warning fix 2020-04-09 20:05:10 +02:00
Francesco Gatti ed0596a52c warning fix 2020-04-09 01:00:14 +02:00
Micaela Verucchi ae1d8cd9e6 Refactoring & documentation 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-04-08 11:20:48 +02:00
Micaela Verucchi 326c7e0940 Move extraction of name into function in utils 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-04-08 10:09:15 +02:00
Micaela Verucchi c36befaf2b Minor fix on output file name in map demo 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-04-08 09:35:54 +02:00
Micaela Verucchi c8308963df Add evaluation to tk::dnn namespace, style fix also 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-04-06 18:12:38 +02:00