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

147 Commits

Author SHA1 Message Date
Francesco Gatti 8d45259b15 SEP model 2020-05-03 15:54:39 +02:00
Micaela Verucchi 3be4fe0117 Modify download 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-04-28 20:40:35 +02:00
Micaela Verucchi 1f4f27d752 Support yolov4 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-04-28 20:35:30 +02:00
Micaela Verucchi fdfda38fbf Merge branch 'master' of https://github.com/ceccocats/tkDNN into cnet 2020-04-28 14:43:24 +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
Micaela Verucchi a761fef159 Add bdd new tests, add BDD100K_val download, fix yolo3-512 link 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-04-24 12:39:57 +02:00
Francesco Gatti 2a6f7db4af Flatten batch to be checked 2020-04-23 01:18:27 +02:00
Francesco Gatti 1cbf18c21e test batch 2020-04-23 00:54:03 +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
Micaela Verucchi ec7620ebb6 Move download of weights inside build folder 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-04-17 22:36:25 +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 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 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