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

87 Commits

Author SHA1 Message Date
Micaela Verucchi 1f4f27d752 Support yolov4 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-04-28 20:35:30 +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 fc873c5219 cmake dont download test data 2020-04-09 20:00:29 +02:00
Francesco Gatti 2389df8626 Merge branch 'master' into cnet 2020-04-09 00:57:43 +02:00
Francesco Gatti 4c5899eae7 remove deprecated and unused warning 2020-04-09 00:55:07 +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 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
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 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
xavier a65c8aa91e Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet 2020-03-12 20:18:11 +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
xavier 98770ee0c5 Merge 
Signed-off-by: xavier <micaelaverucchi@gmail.com>
 2020-03-10 15:43:26 +01:00
xavier 07a61b0e37 Add tiny yolo 512 test 
Signed-off-by: xavier <micaelaverucchi@gmail.com>
 2020-03-10 15:40:11 +01:00
Micaela Verucchi 75d5e20843 Add writing results on file for map demo 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-03-10 14:35:25 +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
Davide Sapienza 99e809737f Move multiple demo files into one 
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-03-03 17:30:35 +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 c876fa05ae LSTM cudnn test 2020-02-13 19:27:18 +01:00
xavier d0026ff235 Change README.md, add config.yaml, refactoring 
Signed-off-by: xavier <micaelaverucchi@gmail.com>
 2020-02-11 11:25:42 +01:00
xavier e01d7c6b13 Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet 2020-02-11 10:32:52 +01:00
xavier 73548b179e Read parameters for mAP from yaml, add yampl-cpp dependency 
Signed-off-by: xavier <micaelaverucchi@gmail.com>
 2020-02-11 10:32:42 +01:00
Davide Sapienza 2b06836a7f Optimize deformable kernel 
Signed-off.by: Ignacio Sañudo Olmedo<ignacio.sanudoolmedo@unimore.it>
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-02-11 10:06:08 +01:00
xavier dc130fc834 Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet 2020-02-05 18:39:32 +01:00
xavier ef18be7b7b Add mAP computation and demo 
Signed-off-by: xavier <micaelaverucchi@gmail.com>
 2020-02-05 18:39:13 +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 b23cf9fe70 Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet 
Signed-off-by: Michaela Verucchi <micaelaverucchi@gmail.com>
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-01-20 12:35:27 +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
xavier 0f0c8c28e1 Merge with master, works with Jetpack 4.3 
Signed-off-by: xavier <micaelaverucchi@gmail.com>
 2020-01-15 19:24:19 +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 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 aa5927d8a1 findCUDNN 2019-11-06 14:04:23 +01:00
Davide Sapienza e9ec582223 Shortcat ok 2019-10-29 15:08:44 +01:00
Micaela Verucchi 4ac2c87d1c cmake fix 2019-10-28 18:39:28 +01:00
Micaela Verucchi 42a1ea02b9 resnet merge 2019-10-28 18:38:23 +01:00
mbosi 92f3d1c548 fixed install cmake 2019-10-01 18:47:31 +02:00
Francesco Gatti bbc4dda635 removed buildtype 2019-09-17 17:16:46 +02:00
Francesco Gatti de8b02fe50 install fix 2019-09-17 16:12:59 +02:00
Francesco Gatti ca62784f57 include dir fix, cmake dir 2019-09-17 15:22:39 +02:00
mbosi a038e966d9 yolo3 flir ok 2019-09-15 16:19:30 +02:00
Francesco Gatti 041968f38a cmake fix 2019-06-29 11:08:30 +02:00
Francesco Gatti f50aa4ad1a fix cmake 2019-06-28 18:51:01 +02:00
Francesco Gatti 6656c3d0e8 fix cmake 2019-06-28 17:44:32 +02:00
rcavicchioli 39f80bbfb6 coco4 2019-02-19 11:38:13 +01:00
Francesco Gatti bdd8e0bc26 yolo3plug fix 2019-02-18 18:55:48 +00:00
Francesco Gatti 0d682136de yolo3 berkeley ok 2019-02-18 15:37:39 +00:00
Francesco Gatti e91db28756 shortcut cu 2018-12-20 09:53:17 +01:00