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

129 Commits

Author SHA1 Message Date
Micaela Verucchi 18aff44c99 Add imuodom download and test in script 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-04-08 16:50:40 +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 e036c5da34 Update download url 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-04-08 12:34:02 +02:00
Micaela Verucchi 065c4e2b58 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 9a1b673db8 Modify weights download for some nets, now also calibration tables are downloaded in tests folder 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-04-06 16:22:23 +02:00
Micaela Verucchi 9e08f4125a Modify weights download link for mobilenetv2ssd512 and yolo3_tiny512 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-04-02 09:41:24 +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 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
Micaela Verucchi f4b976c793 Add tp tests for yolo3512 and yolo3tiny512 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-03-26 15:06:49 +01:00
Micaela Verucchi cae0b84687 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
Micaela Verucchi 1a50065bde Fix conv2d with additional bias for tensorRT. Fix reshape deserialize. Mb2512 works with tensorRT 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-03-17 13:18:05 +01:00
Micaela Verucchi 14d850751d Add mobilenetv2ssdlite512 test. 
Works for CuDNN, not for tensorRT.
Modified channels in second convolution for classification headers from 126 to 486, when changing size from 300 to 512.
Added size 512 SSD specs and support to COCO dataset (81 classes, first BACKGROUND due to repo for training).
Refactoring class names.

Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-03-17 10:35:02 +01:00
Micaela Verucchi 4a2a244162 Add yolov3_512 (size 512) test 2020-03-16 11:29:15 +01:00
Francesco Gatti a7c3761960 eigen assert fix 2020-03-14 19:50:55 +01:00
Micaela Verucchi 94971fbdcb Fix download weights yolov3 versions 
Signed-off-by: Micaela Verucchi <micaela.verucchi@unimore.it>
 2020-03-14 18:20:32 +01:00
Francesco Gatti 677b6b2698 Refactoring 
Signed-off-by: Francesco Gatti <gattifrancesco@hotmail.it>
 2020-03-14 17:22:58 +01:00
Francesco Gatti 21b8146232 Add function to download weigths if do no exist, for each test. Add some controls in map demo. 
Signed-off-by: Francesco Gatti <gattifrancesco@hotmail.it>
 2020-03-14 17:02:07 +01:00
xavier 51b9780baf Add Yolov3_tiny512 test cpp 
Signed-off-by: xavier <micaelaverucchi@gmail.com>
 2020-03-13 12:16:18 +01:00
Francesco Gatti 4845edbc88 csresnext50-panet-spp works with TensorRT 
Signed-off-by: Francesco Gatti <gattifrancesco@hotmail.it>
 2020-03-12 18:24:41 +01:00
Francesco Gatti 5f4508a5a6 Add csresnext50-panet-spp test. Works on CUDNN. Does not work with TensorRT 
Signed-off-by: Francesco Gatti <gattifrancesco@hotmail.it>
 2020-03-11 18:37:26 +01:00
Francesco Gatti 2d378e9070 imuodom fix 2020-03-09 22:05:34 +01:00
Francesco Gatti 6fee8d5ec5 - ImuOdom model into class 
- Route layer input array hard copy
- fix utils
 2020-03-09 19:44:21 +01:00
Micaela Verucchi 1d38d88c85 Fix dependencies problems 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-02-27 16:48:10 +01:00
xavier 3be254b08f Add Mobilenetv2 SSD Lite post and preprocessing, add mobilenet demo 
Signed-off-by: xavier <micaelaverucchi@gmail.com>
 2020-02-26 17:42:20 +01:00
xavier 808a84131c Add Mobilenet2SSDLite test 
The new test works both with TensorRT and cuDNN. Preprocessing and
Postprocessing are missing. Add ClippedReLU (for ReLU6), groups for
Conv2d, additional bias for convolution.

Other minors:
-move the timer in the detector to measure all the
processing time for a given frame (both centernet and yolo);
-add int8 flag.

Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
Davide Sapienza <sapienza.dav@gmail.com>
 2020-02-21 10:45:46 +01:00
Francesco Gatti 4b48a2f38c structure ok, result wrong 2020-02-16 17:08:19 +01:00
Francesco Gatti 0ac292ea48 works but it need cleaning 2020-02-16 16:28:39 +01:00
Francesco Gatti 8736a8c3da LSTM params 2020-02-15 20:37:08 +01:00
Francesco Gatti f5706d8523 lstm return seq 2020-02-13 23:21:28 +01:00
Francesco Gatti c876fa05ae LSTM cudnn test 2020-02-13 19:27:18 +01:00
Davide Sapienza da8ba7d8ce Add Anaconda environment for ResNet101 and DLA34 weights exporter 
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-02-10 10:58:04 +01:00
Davide Sapienza 932de41e38 Add DLA34 weights exporter 
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-02-10 10:56:33 +01:00
Davide Sapienza 5d07e98a9c Update ResNet101 weights exporter 
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-02-10 10:54:04 +01:00
Davide Sapienza 516c1e6636 Change CenterNet input dimension. 
This commit changes the image input dimension, it updates the
CenterNet detection class.

Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-02-06 18:08:06 +01:00
Davide Sapienza 507d30bf52 Implement CenterNet based on DLA34, CUDNN and TensorRT work. 
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-02-05 14:59:56 +01:00
Davide Sapienza 04918f539e Implement DLA34, CUDNN and TensorRT work. 
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-01-24 19:30:32 +01:00
Davide Sapienza 07da54efe9 Centernet: fix pooling problem, add centrnet demo 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-01-21 12:50:18 +01:00
Davide Sapienza 7838cb4922 Pre-process, Process and Post-process work 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-01-20 12:27:49 +01:00
nvidia f0ea2c027f Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet 2019-12-20 15:27:55 +01:00
nvidia 854a4c316a Add ResizeLayerRT plugin 
Signed-off-by: nvidia <micaelaverucchi@gmail.com>
 2019-12-20 15:27:41 +01:00
Davide Sapienza d889ed385d CenterNet TensorRT works. TensorRT serialization not yet implemented 
Signed-oof-by: Davide Sapienza <sapienza.dav@gmail.com>
 2019-12-20 11:05:03 +01:00
Davide Sapienza df888a3457 Add CenterNet based on Resnet101, TensorRT not implemented. 
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2019-12-18 18:59:54 +01:00
Micaela Verucchi 44b2bce3ff Yolo3_tiny CUDNN works, TensorRT doesn't. Add n_masks to Yolo layer. 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2019-12-05 20:43:48 +00:00
Micaela Verucchi a2db98670a Add Yolov3 (COCO80) and Yolov3-tiny (COCO80), TensorRT for tiny not working 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2019-12-04 17:01:40 +00:00
Francesco Gatti 2594f59d0d conv2d ok, but deconv ha different dim with tensorrt 2019-10-30 16:45:46 +01:00
Francesco Gatti f247300469 test simple 2019-10-30 15:40:48 +01:00
Davide Sapienza 02936fa928 ResNet working 2019-10-29 17:21:31 +01:00
Davide Sapienza e9ec582223 Shortcat ok 2019-10-29 15:08:44 +01:00
Micaela Verucchi 058f9b795f resnet weight export fix 2019-10-28 19:20:02 +01:00
Micaela Verucchi 42a1ea02b9 resnet merge 2019-10-28 18:38:23 +01:00