 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 |
|
|
xavier
|
57d7743f7e
|
Change opencv funcion call (due to OpenCV 4)
Signed-off-by: xavier <micaelaverucchi@gmail.com>
|
2020-01-15 09:55:10 +01:00 |
|
 fbagni
|
23a1365dc4
|
Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet
|
2019-12-23 16:14:24 +01:00 |
|
|
fbagni
|
7e7f480e2b
|
Yolov3_tiny works on tensorRT
Signed-off-by: fbagni <gattinomicino>
|
2019-12-23 16:11:41 +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 |
|
 mbosi
|
6bf9179acc
|
fix to drivework global path
|
2019-12-12 12:30:24 +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
|
b218b18a02
|
readme update
|
2019-12-02 20:24:12 +01:00 |
|
|
Francesco Gatti
|
aa5927d8a1
|
findCUDNN
|
2019-11-06 14:04:23 +01: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 |
|
|
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
|
d6d93a74f8
|
fix
|
2019-10-30 09:42:54 +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 |
|
|
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
|
4ac2c87d1c
|
cmake fix
|
2019-10-28 18:39:28 +01:00 |
|
|
Micaela Verucchi
|
42a1ea02b9
|
resnet merge
|
2019-10-28 18:38:23 +01:00 |
|
|
mbosi
|
92f3d1c548
|
fixed install cmake
|
2019-10-01 18:47:31 +02:00 |
|
|
Francesco Gatti
|
bbc4dda635
|
removed buildtype
|
2019-09-17 17:16:46 +02:00 |
|
|
Francesco Gatti
|
de8b02fe50
|
install fix
|
2019-09-17 16:12:59 +02: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 |
|
|
Francesco Gatti
|
77f031c0f4
|
save video result
|
2019-09-16 10:35:29 +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 |
|
|
Francesco Gatti
|
041968f38a
|
cmake fix
|
2019-06-29 11:08:30 +02:00 |
|
|
Francesco Gatti
|
f50aa4ad1a
|
fix cmake
|
2019-06-28 18:51:01 +02:00 |
|
|
Francesco Gatti
|
6656c3d0e8
|
fix cmake
|
2019-06-28 17:44:32 +02:00 |
|
 Autochaffeur
|
4ebbb6af2b
|
README update
|
2019-05-13 17:32:18 +02:00 |
|
|
mbosi
|
eef1fd321f
|
added label to demo bounding box visualization
|
2019-05-02 14:30:50 +02:00 |
|
|
Francesco Gatti
|
a85367fa22
|
dla commented
|
2019-03-07 17:40:24 +01:00 |
|
 Roberto Cavicchioli
|
3714155809
|
dla
|
2019-03-06 16:40:12 +01:00 |
|
|
Francesco Gatti
|
c22219ad16
|
DLA number print
|
2019-03-06 13:02:39 +01:00 |
|
|
Francesco Gatti
|
7505c28d2d
|
include fix
|
2019-02-19 11:09:27 +00:00 |
|
|
rcavicchioli
|
39f80bbfb6
|
coco4
|
2019-02-19 11:38:13 +01:00 |
|
|
rcavicchioli
|
851c6a366c
|
arg fix
|
2019-02-19 11:10:34 +01:00 |
|
|
Francesco Gatti
|
de04ae1cab
|
doc
|
2019-02-19 09:03:33 +00:00 |
|
|
Francesco Gatti
|
1aa4f0275d
|
color fix
|
2019-02-19 08:57:51 +00:00 |
|
|
Francesco Gatti
|
c7941666ec
|
demo for more yolo3
|
2019-02-19 08:43:35 +00:00 |
|