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

146 Commits

Author SHA1 Message Date
Micaela Verucchi adb5a693cd Modify download 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-04-28 20:40:35 +02:00
Micaela Verucchi db0f8a4d99 Support yolov4 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-04-28 20:35:30 +02:00
Micaela Verucchi a874fad2bd Merge branch 'master' of https://github.com/ceccocats/tkDNN into cnet 2020-04-28 14:43:24 +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
Micaela Verucchi f418a1f9b2 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 8cff886ee5 Flatten batch to be checked 2020-04-23 01:18:27 +02:00
Francesco Gatti 488887992c test batch 2020-04-23 00:54:03 +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
Micaela Verucchi 2fbac7705d 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 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 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 bddb0110ca 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 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 da79128271 Update download url 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-04-08 12:34:02 +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 dfdeb8b36e 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 79a05a8159 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 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 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
Micaela Verucchi 7c55dcb708 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 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
Micaela Verucchi 43fd92f701 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 1d388ca51a 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 9c25d15ff2 Add yolov3_512 (size 512) test 2020-03-16 11:29:15 +01:00
Francesco Gatti 62c8f528a4 eigen assert fix 2020-03-14 19:50:55 +01:00
Micaela Verucchi 82d907971c Fix download weights yolov3 versions 
Signed-off-by: Micaela Verucchi <micaela.verucchi@unimore.it>
 2020-03-14 18:20:32 +01:00
Francesco Gatti 555b32b5dc Refactoring 
Signed-off-by: Francesco Gatti <gattifrancesco@hotmail.it>
 2020-03-14 17:22:58 +01:00
Francesco Gatti cf7fbadcd2 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 32dd807ab0 Add Yolov3_tiny512 test cpp 
Signed-off-by: xavier <micaelaverucchi@gmail.com>
 2020-03-13 12:16:18 +01:00
Francesco Gatti 13b4dffd71 csresnext50-panet-spp works with TensorRT 
Signed-off-by: Francesco Gatti <gattifrancesco@hotmail.it>
 2020-03-12 18:24:41 +01:00
Francesco Gatti f6527f51e3 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 8944778dcf imuodom fix 2020-03-09 22:05:34 +01:00
Francesco Gatti 406f8cc9b3 - ImuOdom model into class 
- Route layer input array hard copy
- fix utils
 2020-03-09 19:44:21 +01:00
Micaela Verucchi fe206ea24c Fix dependencies problems 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-02-27 16:48:10 +01:00
xavier 3eb079dd13 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 38a1b9dcb2 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 1a1c54f364 structure ok, result wrong 2020-02-16 17:08:19 +01:00
Francesco Gatti 10b7160677 works but it need cleaning 2020-02-16 16:28:39 +01:00
Francesco Gatti 4746121d43 LSTM params 2020-02-15 20:37:08 +01:00
Francesco Gatti 4fa5d2c231 lstm return seq 2020-02-13 23:21:28 +01:00
Francesco Gatti a9c0db0bf6 LSTM cudnn test 2020-02-13 19:27:18 +01:00
Davide Sapienza c02238ddc8 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 e72aa348a0 Add DLA34 weights exporter 
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-02-10 10:56:33 +01:00
Davide Sapienza 62fe82ce9e Update ResNet101 weights exporter 
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-02-10 10:54:04 +01:00