98770ee0c5
Merge
...
Signed-off-by: xavier <micaelaverucchi@gmail.com >
2020-03-10 15:43:26 +01:00
07a61b0e37
Add tiny yolo 512 test
...
Signed-off-by: xavier <micaelaverucchi@gmail.com >
2020-03-10 15:40:11 +01:00
1d38d88c85
Fix dependencies problems
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-02-27 16:48:10 +01:00
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
c4cef1aa02
Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet
2020-02-21 10:49:35 +01:00
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
200e9466f6
Move pre-processing on GPU
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-02-14 18:44:13 +01:00
17a083ae28
Remove mallocs and frees from the kernels
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-02-07 14:50:05 +01:00
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
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
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
Davide Sapienza