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

174 Commits

Author SHA1 Message Date
Micaela Verucchi 360adbd1a1 Add verbose define 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-05-15 09:41:27 +02:00
Micaela Verucchi 7df266cac4 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 b8686186e4 Merge branch 'master' of https://github.com/ceccocats/tkDNN into eval 2020-05-11 11:58:51 +02:00
Micaela Verucchi 5eb0cbfcef 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 54074780cf filter yaw ok 2020-05-03 19:12:10 +02:00
Francesco Gatti 3fcb05be64 Merge branch 'tree' into ipmslam 2020-05-03 15:58:43 +02:00
Francesco Gatti 8d45259b15 SEP model 2020-05-03 15:54:39 +02:00
Francesco Gatti bdace759a9 merge tree 2020-05-01 15:39:00 +02:00
Micaela Verucchi 1f4f27d752 Support yolov4 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-04-28 20:35:30 +02:00
Francesco Gatti f91fb7f287 stream 2020-04-28 14:36:02 +02:00
Micaela Verucchi dbe54f4b67 Fix shortcutRT plugin, now works with batches, fix yolo3_512 downlaod link 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-04-28 14:34:12 +02:00
Francesco Gatti 3aa3097bbb sensor close if not started fix 2020-04-28 12:59:30 +02:00
dsapienza 26d8d009c5 Deformable batch works 
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-04-25 02:47:31 +02:00
Francesco Gatti 6a185564e7 RegionRT softmax fix 2020-04-24 00:08:06 +02:00
Francesco Gatti afaeb75d84 Reshape batch fix 2020-04-23 23:51:23 +02:00
Francesco Gatti 2a6f7db4af Flatten batch to be checked 2020-04-23 01:18:27 +02:00
Francesco Gatti 9655c29b94 RouteRt is not used 2020-04-21 19:50:45 +00:00
Francesco Gatti a91b2f46ba batch seems ok in yolo3_berkely 
layers to be checked:
DeformableConvRT
FlattenConcatRT
ReshapeRT
RouteRT (dont know why but seems working)
 2020-04-21 19:41:40 +00:00
Francesco Gatti ef91439c87 batch size > 1 2020-04-21 19:20:44 +02:00
Francesco Gatti 585f66e2f6 warning fix 2020-04-09 20:05:10 +02:00
Davide Sapienza 50ada3db60 Remove the optional parameters from the Pooling layer. 
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-04-09 19:20:27 +02:00
Micaela Verucchi a8fd4cabf3 Add mnist download, change ERROR_TKDNN into ERROR_TENSORRT 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-04-09 19:02:25 +02:00
Davide Sapienza 8a575f6447 Remove the final parameter from the layers 
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-04-09 18:48:03 +02:00
Micaela Verucchi af761786ea Fix pooling, add return code in each test, add return code handling in test_all_tests script 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-04-09 15:25:05 +02:00
Francesco Gatti c248a94101 warning fix 2020-04-09 01:00:14 +02:00
Francesco Gatti 2389df8626 Merge branch 'master' into cnet 2020-04-09 00:57:43 +02:00
Francesco Gatti 73fb12be0f gnuplot imuodom 2020-04-08 16:53:35 +02:00
Micaela Verucchi 8c1eb69494 Merge with master works 
Signed-off-by:  Micaela Verucchi <micaelaverucchi@gmail.com>
		Davide Sapienza <sapienza.dav@gmail.com>
 2020-04-08 14:59:42 +02:00
Micaela Verucchi 34968eb4ad Add BoundingBox class 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-04-08 13:52:29 +02:00
Micaela Verucchi b070bb3bd8 Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet 2020-04-08 11:20:56 +02:00
Micaela Verucchi 6a08118e3d Refactoring & documentation 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-04-08 11:20:48 +02:00
Davide Sapienza 5be71cf9b7 Fix the Deformable convolution code sintax. 
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-04-08 11:14:34 +02:00
Davide Sapienza 3a5115578e Fix the INT8 calibrator sintax 
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-04-08 10:22:03 +02:00
Micaela Verucchi 6e55e2376f Move extraction of name into function in utils 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-04-08 10:09:15 +02:00
Davide Sapienza 537b0cfa21 Fix the sorting kernels used in the CenterNet pre and post-processing. 
This commit moves the kernels in the correct sub-directory.
It creates new header file for Thrust kernels. It splits the
kernels into two files: 'normalize.cu' contains CenterNet
pre-processing operations, 'postprocessing.cu' contains the
CenterNet post-processing operations.

Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-04-07 18:28:12 +02:00
Davide Sapienza 52e1a192e3 Fix the tensorRT code to support versions prior to 6.0 
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-04-06 19:54:40 +02:00
Micaela Verucchi c63ff6ecd2 Fix compile errors dut to Int8BatchStream 
Signed-off-by: Micaela Verucchi <micaela.verucchi@unimore.it>
               Davide Sapienza <sapienza.dav@gmail.com>
 2020-04-06 19:11:39 +02:00
Micaela Verucchi cddd420942 Add again evalutation.h 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-04-06 18:35:52 +02:00
Micaela Verucchi 769284574b Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet 2020-04-06 18:13:02 +02:00
Micaela Verucchi 733755e7e0 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
Davide Sapienza 2fbd24625c Move the calibration table saving to the tests/network folder. 
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-04-06 18:06:21 +02:00
Micaela Verucchi 92dae7f5c5 Add opencv script, modified macro, add opencv section into readme 
Signed-off-by: Micaela Verucchi <micaela.verucchi@unimore.it>
 2020-04-06 17:25:53 +02:00
Micaela Verucchi 43567dc3ea Add getMemoryUsage function, detection update moved in abstract lass, splitted execution time in pre-inf-post, other minors. 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-04-01 19:18:43 +02:00
Davide Sapienza 03d2b88fa6 Add TKDNN_CALIB_IMG_PATH and TKDNN_CALIB_LABRL_PATH variable 
This commit adds two variables for the calibration dataset.
The first is reffered to .txt file that contains the list
of the absolute paths of the images for the INT8 calitration.
The second is referred to .txt file that contains the list
of the absolute paths of the labels of the same images above.

Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-03-30 18:55:32 +02:00
Davide Sapienza 0474f019cf Add TKDNN_MODE variable to the name of the network. 
This commit permits to obtain different .rt files for different
precision optimizations of the same network.

Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-03-30 15:08:25 +02:00
Davide Sapienza e217a0e83f Fix opencv cuda include error 
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-03-27 10:43:49 +01:00
Davide Sapienza e540213da6 Add the Int8 calibrator and the tensorRT Int8 inference 
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-03-27 00:48:14 +01:00
Micaela Verucchi 785a3d8dd5 Add GPU version for yolo3detection 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-03-23 21:29:29 +01:00
Micaela Verucchi 3bc648c07b Modify map demo, using abstract class. Move draw function in abstract class 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-03-23 19:32:37 +01:00
Micaela Verucchi 8b9516da97 Refactoring for detection NN 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-03-20 21:14:12 +01:00