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
7c55dcb708
Add tp tests for yolo3512 and yolo3tiny512
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-03-26 15:06:49 +01:00
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
9c25d15ff2
Add yolov3_512 (size 512) test
2020-03-16 11:29:15 +01:00
9d17cb42f4
Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet
2020-03-12 20:18:11 +01:00
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
d0d462c015
Merge
...
Signed-off-by: xavier <micaelaverucchi@gmail.com >
2020-03-10 15:43:26 +01:00
2d46f16b70
Add tiny yolo 512 test
...
Signed-off-by: xavier <micaelaverucchi@gmail.com >
2020-03-10 15:40:11 +01:00
4b85a2238a
Add writing results on file for map demo
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-03-10 14:35:25 +01:00
01a42ffe27
Move multiple demo files into one
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-03-03 17:30:35 +01:00
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
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
1936e54870
Change README.md, add config.yaml, refactoring
...
Signed-off-by: xavier <micaelaverucchi@gmail.com >
2020-02-11 11:25:42 +01:00
b5c0baa99d
Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet
2020-02-11 10:32:52 +01:00
801b8b5641
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
51ffcb1f50
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
37019597af
Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet
2020-02-05 18:39:32 +01:00
d8eeb36d4b
Add mAP computation and demo
...
Signed-off-by: xavier <micaelaverucchi@gmail.com >
2020-02-05 18:39:13 +01:00
c695d8c5d7
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
fe85c26888
Implement DLA34, CUDNN and TensorRT work.
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-01-24 19:30:32 +01:00
7f239efdc0
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
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
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
0f0c8c28e1
Merge with master, works with Jetpack 4.3
...
Signed-off-by: xavier <micaelaverucchi@gmail.com >
2020-01-15 19:24:19 +01:00
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
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
aa5927d8a1
findCUDNN
2019-11-06 14:04:23 +01:00
e9ec582223
Shortcat ok
2019-10-29 15:08:44 +01:00
4ac2c87d1c
cmake fix
2019-10-28 18:39:28 +01:00
42a1ea02b9
resnet merge
2019-10-28 18:38:23 +01:00
92f3d1c548
fixed install cmake
2019-10-01 18:47:31 +02:00
bbc4dda635
removed buildtype
2019-09-17 17:16:46 +02:00
de8b02fe50
install fix
2019-09-17 16:12:59 +02:00
ca62784f57
include dir fix, cmake dir
2019-09-17 15:22:39 +02:00
a038e966d9
yolo3 flir ok
2019-09-15 16:19:30 +02:00
041968f38a
cmake fix
2019-06-29 11:08:30 +02:00
f50aa4ad1a
fix cmake
2019-06-28 18:51:01 +02:00
6656c3d0e8
fix cmake
2019-06-28 17:44:32 +02:00
39f80bbfb6
coco4
2019-02-19 11:38:13 +01:00
bdd8e0bc26
yolo3plug fix
2019-02-18 18:55:48 +00:00
0d682136de
yolo3 berkeley ok
2019-02-18 15:37:39 +00:00
e91db28756
shortcut cu
2018-12-20 09:53:17 +01:00
bc0ea65766
yolo3 debug start
2018-12-19 19:39:31 +01:00
6eb63160c8
berkeley
2018-12-18 14:52:18 +01:00
443179359d
config
2018-12-03 22:04:04 +01:00
a13bc2f007
../CMakeLists.txt
2018-12-03 17:52:44 +01:00
4d30f0abd7
compile on x86
2018-12-03 17:37:24 +01:00
09679d7bb6
voc
2018-09-18 16:27:09 +02:00
6331724953
live detection
2018-09-15 08:57:43 +00:00
2cf8d8f6fc
fp16 implementation, TODO deallocate in LayerWgs
2017-08-30 14:37:25 +00:00
Micaela Verucchi
Francesco Gatti
Davide Sapienza
mbosi
rcavicchioli
Alessio