0474f019cf
Add TKDNN_MODE variable to the name of the network.
...
This commit permits to obtain different .rt files for different
precision optimizations of the same network.
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-03-30 15:08:25 +02:00
f4b976c793
Add tp tests for yolo3512 and yolo3tiny512
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-03-26 15:06:49 +01:00
cae0b84687
Add preprocess function, allow preprocess on GPU for mobilenetdetection
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-03-19 17:53:29 +01:00
1a50065bde
Fix conv2d with additional bias for tensorRT. Fix reshape deserialize. Mb2512 works with tensorRT
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-03-17 13:18:05 +01:00
14d850751d
Add mobilenetv2ssdlite512 test.
...
Works for CuDNN, not for tensorRT.
Modified channels in second convolution for classification headers from 126 to 486, when changing size from 300 to 512.
Added size 512 SSD specs and support to COCO dataset (81 classes, first BACKGROUND due to repo for training).
Refactoring class names.
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-03-17 10:35:02 +01:00
4a2a244162
Add yolov3_512 (size 512) test
2020-03-16 11:29:15 +01:00
94971fbdcb
Fix download weights yolov3 versions
...
Signed-off-by: Micaela Verucchi <micaela.verucchi@unimore.it >
2020-03-14 18:20:32 +01:00
677b6b2698
Refactoring
...
Signed-off-by: Francesco Gatti <gattifrancesco@hotmail.it >
2020-03-14 17:22:58 +01:00
21b8146232
Add function to download weigths if do no exist, for each test. Add some controls in map demo.
...
Signed-off-by: Francesco Gatti <gattifrancesco@hotmail.it >
2020-03-14 17:02:07 +01:00
51b9780baf
Add Yolov3_tiny512 test cpp
...
Signed-off-by: xavier <micaelaverucchi@gmail.com >
2020-03-13 12:16:18 +01:00
4845edbc88
csresnext50-panet-spp works with TensorRT
...
Signed-off-by: Francesco Gatti <gattifrancesco@hotmail.it >
2020-03-12 18:24:41 +01:00
5f4508a5a6
Add csresnext50-panet-spp test. Works on CUDNN. Does not work with TensorRT
...
Signed-off-by: Francesco Gatti <gattifrancesco@hotmail.it >
2020-03-11 18:37:26 +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
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
da8ba7d8ce
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
932de41e38
Add DLA34 weights exporter
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-02-10 10:56:33 +01:00
5d07e98a9c
Update ResNet101 weights exporter
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-02-10 10:54:04 +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
507d30bf52
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
04918f539e
Implement DLA34, CUDNN and TensorRT work.
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-01-24 19:30:32 +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
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
Davide Sapienza
Micaela Verucchi
Micaela Verucchi
Francesco Gatti
mbosi
rcavicchioli