Commit Graph

73 Commits

Author SHA1 Message Date
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
Francesco Gatti 57c9a6ec99 LEAKY serialized 2017-08-11 16:32:22 +02:00
Francesco Gatti 3124f86878 stream in TRT plugin 2017-08-10 19:21:15 +02:00
Francesco Gatti 266330009c opencv viz 2017-08-10 16:22:17 +02:00
Francesco Gatti d7ce952465 get regions 2017-08-08 17:17:24 +02:00
Francesco Gatti 0a9957ba18 network print 2017-08-08 14:59:25 +02:00
Francesco Gatti 7d570c0df4 tiny yolo not working 2017-08-07 15:05:48 +02:00