a194ba59f6
Fix typos ( #107 )
...
Signed-off-by: micaela <micaelaverucchi@gmail.com >
2020-09-11 09:13:59 +02:00
8fad424c61
Yolo4-tiny batched
...
fix #59
2020-06-30 19:37:16 +02:00
4c9b9bf03a
yolo4tiny works on tensorRT 🐬 🐬 🐬 🐬
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-06-30 15:52:58 +02:00
1f4f27d752
Support yolov4
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-28 20:35:30 +02:00
f91fb7f287
stream
2020-04-28 14:36:02 +02:00
dbe54f4b67
Fix shortcutRT plugin, now works with batches, fix yolo3_512 downlaod link
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-28 14:34:12 +02:00
26d8d009c5
Deformable batch works
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-04-25 02:47:31 +02:00
6a185564e7
RegionRT softmax fix
2020-04-24 00:08:06 +02:00
afaeb75d84
Reshape batch fix
2020-04-23 23:51:23 +02:00
2a6f7db4af
Flatten batch to be checked
2020-04-23 01:18:27 +02:00
9655c29b94
RouteRt is not used
2020-04-21 19:50:45 +00:00
a91b2f46ba
batch seems ok in yolo3_berkely
...
layers to be checked:
DeformableConvRT
FlattenConcatRT
ReshapeRT
RouteRT (dont know why but seems working)
2020-04-21 19:41:40 +00:00
ef91439c87
batch size > 1
2020-04-21 19:20:44 +02:00
5be71cf9b7
Fix the Deformable convolution code sintax.
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-04-08 11:14:34 +02:00
4845edbc88
csresnext50-panet-spp works with TensorRT
...
Signed-off-by: Francesco Gatti <gattifrancesco@hotmail.it >
2020-03-12 18:24:41 +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
5abf1613f2
Fix memory leak
...
This commit moves cublasCreate out from dcn_v2_cuda_forward
to save some milliseconds and it adds cublasDestroy (cause
of memory leak).
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-02-06 23:09:12 +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
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
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
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
ca62784f57
include dir fix, cmake dir
2019-09-17 15:22:39 +02:00
micaela
Francesco Gatti
dsapienza
Davide Sapienza