 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 |
|
|
xavier
|
c32a0be257
|
Batchnorm eps fix, works on jetpack 4.3
|
2020-01-15 18:06:02 +01:00 |
|
 Davide Sapienza
|
5272f1cde6
|
CenterNet TensorRT serialization works
This commit adds the Deformable layer serialization.
Signed-oof-by: Davide Sapienza <sapienza.dav@gmail.com>
|
2019-12-23 15:31:13 +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
|
e99b353d8b
|
Fix the inference operation of the deformable convolutional layer.
This commit removes the malloc operation in the inference
method and adds the sigmoid kernel.
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
|
2019-12-19 14:47:57 +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 |
|
 Francesco Gatti
|
2594f59d0d
|
conv2d ok, but deconv ha different dim with tensorrt
|
2019-10-30 16:45:46 +01:00 |
|
|
Micaela Verucchi
|
6a1707f65e
|
Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet
|
2019-10-30 10:20:44 +01:00 |
|
|
Micaela Verucchi
|
8b64d1876f
|
deformable conv cu
|
2019-10-30 10:20:04 +01:00 |
|
 fbagni
|
33f36ab204
|
Deconv tensorrt
|
2019-10-30 09:40:29 +01:00 |
|
|
Francesco Gatti
|
f9afee2f3b
|
deconv layer cudnn
|
2019-10-30 00:05:22 +01:00 |
|
|
Davide Sapienza
|
02936fa928
|
ResNet working
|
2019-10-29 17:21:31 +01:00 |
|
|
Micaela Verucchi
|
42a1ea02b9
|
resnet merge
|
2019-10-28 18:38:23 +01:00 |
|
|
Francesco Gatti
|
ca62784f57
|
include dir fix, cmake dir
|
2019-09-17 15:22:39 +02:00 |
|
|
Francesco Gatti
|
ec02c7292f
|
save layer names in rt file
|
2019-09-16 19:41:59 +02:00 |
|
 mbosi
|
8c629ebe7b
|
string input and flir test
|
2019-09-14 19:03:13 +02:00 |
|
 Roberto Cavicchioli
|
3714155809
|
dla
|
2019-03-06 16:40:12 +01:00 |
|
|
Francesco Gatti
|
7505c28d2d
|
include fix
|
2019-02-19 11:09:27 +00:00 |
|
|
Francesco Gatti
|
c7941666ec
|
demo for more yolo3
|
2019-02-19 08:43:35 +00: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
|
738fa94150
|
version update
|
2019-02-18 15:54:22 +00:00 |
|
|
Francesco Gatti
|
13063b904d
|
yolo3 ok
|
2019-02-18 15:51:57 +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
|
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
|
64626bf547
|
shortcut rt test
|
2018-12-21 15:53:39 +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
|
7a46601306
|
yolo3 layer 15
|
2018-12-20 11:36:10 +01:00 |
|
|
Francesco Gatti
|
e91db28756
|
shortcut cu
|
2018-12-20 09:53:17 +01:00 |
|
|
Francesco Gatti
|
ed02930464
|
upsample template
|
2018-12-19 22:45:43 +01:00 |
|
|
Francesco Gatti
|
5f25e0b5f6
|
shortcut template
|
2018-12-19 22:36:46 +01:00 |
|
|
Francesco Gatti
|
6249956469
|
namespace change
|
2018-12-14 21:55:16 +01:00 |
|
|
Francesco Gatti
|
2cf8d8f6fc
|
fp16 implementation, TODO deallocate in LayerWgs
|
2017-08-30 14:37:25 +00:00 |
|
|
Francesco Gatti
|
b2d6dcd207
|
detect demo with mAP
|
2017-08-29 16:48:18 +00:00 |
|
|
Francesco Gatti
|
168a1d8b27
|
color
|
2017-08-25 06:09:29 -07:00 |
|
|
Francesco Gatti
|
0119b31455
|
class in box
|
2017-08-22 06:37:43 -07:00 |
|
|
Francesco Gatti
|
c41a0a09a6
|
version fix
|
2017-08-22 01:43:36 -07:00 |
|
|
Francesco Gatti
|
5595b8037b
|
interpret
|
2017-08-22 01:32:59 -07:00 |
|
|
Francesco Gatti
|
0aa9de4ce8
|
better rt inference
|
2017-08-21 12:10:17 +00:00 |
|
|
Francesco Gatti
|
b3a369dc29
|
RTinference test
|
2017-08-14 11:24:23 +02:00 |
|
|
Francesco Gatti
|
3b2f062dd9
|
tensorRT serialization OK
|
2017-08-11 17:17:05 +02:00 |
|