fe206ea24c
Fix dependencies problems
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-02-27 16:48:10 +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
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
e72aa348a0
Add DLA34 weights exporter
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-02-10 10:56:33 +01:00
62fe82ce9e
Update ResNet101 weights exporter
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-02-10 10:54:04 +01:00
70eb5214cc
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
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
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
f0ea2c027f
Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet
2019-12-20 15:27:55 +01:00
854a4c316a
Add ResizeLayerRT plugin
...
Signed-off-by: nvidia <micaelaverucchi@gmail.com >
2019-12-20 15:27:41 +01:00
d889ed385d
CenterNet TensorRT works. TensorRT serialization not yet implemented
...
Signed-oof-by: Davide Sapienza <sapienza.dav@gmail.com >
2019-12-20 11:05:03 +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
44b2bce3ff
Yolo3_tiny CUDNN works, TensorRT doesn't. Add n_masks to Yolo layer.
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2019-12-05 20:43:48 +00: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
2594f59d0d
conv2d ok, but deconv ha different dim with tensorrt
2019-10-30 16:45:46 +01:00
f247300469
test simple
2019-10-30 15:40:48 +01:00
02936fa928
ResNet working
2019-10-29 17:21:31 +01:00
e9ec582223
Shortcat ok
2019-10-29 15:08:44 +01:00
058f9b795f
resnet weight export fix
2019-10-28 19:20:02 +01:00
42a1ea02b9
resnet merge
2019-10-28 18:38:23 +01:00
ec02c7292f
save layer names in rt file
2019-09-16 19:41:59 +02:00
a038e966d9
yolo3 flir ok
2019-09-15 16:19:30 +02:00
8c629ebe7b
string input and flir test
2019-09-14 19:03:13 +02:00
39f80bbfb6
coco4
2019-02-19 11:38:13 +01:00
87fe342ca2
yoloRT load anchors
2019-02-18 21:39:14 +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
2c63bf05be
multipl yolo morge
2019-02-06 22:24:01 +00:00
0e97452460
dects dont works
2019-02-05 20:09:47 +00:00
c8dea4668d
compute detections
2019-02-04 20:34:15 +00:00
88097a3774
yolo3 ok
2019-01-04 22:28:10 +01:00
2e8d0b1002
yolo3 86 route error
2018-12-23 16:20:17 +01:00
3bd725801d
upsample ok, route have problems
2018-12-22 23:56:01 +01:00
34be4cd00f
yoloRT layer
2018-12-22 21:26:50 +01:00
3b60de00f8
2 input shortcut
2018-12-21 16:17:48 +01:00
64626bf547
shortcut rt test
2018-12-21 15:53:39 +01:00
7a51b4382d
yolo3 ok
2018-12-21 15:28:47 +01:00
c13bda3863
yolo layer break everything
2018-12-21 11:07:35 +01:00
2606820300
layer 96 dont match
2018-12-20 18:17:40 +01:00
a41b22e1f2
layer 94
2018-12-20 17:35:49 +01:00
c8f2e1b448
upsample ok
2018-12-20 17:08:31 +01:00
2ab47b5874
yolo layer
2018-12-20 16:10:01 +01:00
67cc566a0d
layer 81
2018-12-20 14:52:22 +01:00
217ff20058
layer 61
2018-12-20 12:02:47 +01:00
991abdb410
layer 36
2018-12-20 11:46:58 +01:00
7a46601306
yolo3 layer 15
2018-12-20 11:36:10 +01:00
bc0ea65766
yolo3 debug start
2018-12-19 19:39:31 +01:00
Micaela Verucchi
Davide Sapienza
Francesco Gatti
mbosi
rcavicchioli