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

171 Commits

Author SHA1 Message Date
Francesco Gatti 4e1c7a70b1 darknet parser interface 2020-05-30 16:58:40 +02:00
Francesco Gatti ca7631250c filter yaw ok 2020-05-03 19:12:10 +02:00
Francesco Gatti 7f0d9a930d Merge branch 'tree' into ipmslam 2020-05-03 15:58:43 +02:00
Francesco Gatti 5a9ed44b6a SEP model 2020-05-03 15:54:39 +02:00
Francesco Gatti 98537624cf merge tree 2020-05-01 15:39:00 +02:00
Micaela Verucchi db0f8a4d99 Support yolov4 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-04-28 20:35:30 +02:00
Francesco Gatti 04bd5d7ff4 stream 2020-04-28 14:36:02 +02:00
Micaela Verucchi 89f91f568e 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 c975a467b1 sensor close if not started fix 2020-04-28 12:59:30 +02:00
dsapienza 6e5ab031c4 Deformable batch works 
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-04-25 02:47:31 +02:00
Francesco Gatti 13c9dc6620 RegionRT softmax fix 2020-04-24 00:08:06 +02:00
Francesco Gatti 1ea6097a97 Reshape batch fix 2020-04-23 23:51:23 +02:00
Francesco Gatti 8cff886ee5 Flatten batch to be checked 2020-04-23 01:18:27 +02:00
Francesco Gatti 8caff5f598 RouteRt is not used 2020-04-21 19:50:45 +00:00
Francesco Gatti 3d940a9fa2 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 7c81c5a43c batch size > 1 2020-04-21 19:20:44 +02:00
Francesco Gatti 85588f7343 warning fix 2020-04-09 20:05:10 +02:00
Davide Sapienza 106cbb5c73 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 fabad7bace 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 4361d5fec0 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 3502c5b676 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 ed0596a52c warning fix 2020-04-09 01:00:14 +02:00
Francesco Gatti 2635d24855 Merge branch 'master' into cnet 2020-04-09 00:57:43 +02:00
Francesco Gatti c1247930c9 gnuplot imuodom 2020-04-08 16:53:35 +02:00
Micaela Verucchi 0951f4644f 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 ef78d7f176 Add BoundingBox class 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-04-08 13:52:29 +02:00
Micaela Verucchi 5f6011b206 Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet 2020-04-08 11:20:56 +02:00
Micaela Verucchi ae1d8cd9e6 Refactoring & documentation 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-04-08 11:20:48 +02:00
Davide Sapienza 0e5c90634f 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 5562f599a6 Fix the INT8 calibrator sintax 
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-04-08 10:22:03 +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
Davide Sapienza cf3fbeddbd 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 26dcb7b8c9 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 455291b6dd 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 8575059666 Add again evalutation.h 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-04-06 18:35:52 +02:00
Micaela Verucchi 9ec0913d7f Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet 2020-04-06 18:13:02 +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
Davide Sapienza e3442e4764 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 674c61d280 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 af13e7c954 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 bb157be82c 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 ce20868bad 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 482e122655 Fix opencv cuda include error 
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-03-27 10:43:49 +01:00
Davide Sapienza 08e1801c60 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 3eda9b9219 Add GPU version for yolo3detection 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-03-23 21:29:29 +01:00
Micaela Verucchi df37e11709 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 bbcc33c0cf Refactoring for detection NN 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-03-20 21:14:12 +01:00
Micaela Verucchi c7d9c38ea0 Add OPENCV_CUDA define, to allow having preprocess both in CPU and GPU (Mobilenet and Centernet) 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-03-20 10:21:11 +01:00
Micaela Verucchi 58fe723b6a Style fix, useless params removed 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-03-19 18:13:35 +01:00
Micaela Verucchi f44f377771 Add preprocess function, allow preprocess on GPU for mobilenetdetection 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-03-19 17:53:29 +01:00