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