mmr/tkDNN
Archived
1
0
Fork 0
Code Issues 46 Pull Requests 12 Releases 4 Activity
442 Commits 11 Branches 5 Tags
6611a912012d382b9b04c10e105c3943d4ac18f0
Commit Graph

82 Commits

Author SHA1 Message Date
Francesco Gatti 6611a91201 merge perseusdg windows10 support 2021-04-24 18:55:26 +02:00
perseusdg 1de804f98d Code cleanup and readme fixes 2021-04-09 13:17:41 +05:30
perseusdg cc594f09ef merge from gitlab 2021-04-08 00:28:00 +05:30
hchandirasekar f12ec3c935 added yolo4x and yolo4-csp from the github repo and download file corrections 2021-03-26 12:18:46 +05:30
hchandirasekar e94e1f7622 tkdnn first patch for windows 2021-03-24 22:26:46 +05:30
Harshvardhan Chandirasekar 06787a931f minor migrations 2021-03-17 23:06:50 +05:30
Harshvardhan Chandirasekar 304ab49897 shared_ptr migrations 2021-03-17 22:26:26 +05:30
Micaela Verucchi adac8576b0 Add support to Scaled-YOLO4, update Yolov4x-mish (tested) 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2021-01-22 17:57:42 +01:00
Micaela Verucchi 702791e41a Add support for yolov4x-mish. 
Changes:
- add parameters nms_kind, nms_thresh, new_coords to yolo layer and darknet parser
- added diou nms, new method to compute the BBs
- created test for yolov4x-mish called yolo4x

Tested, all tests work.
Problem to solve: little loss in mAP of yolo4x

Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-11-23 11:29:32 +01:00
micaela fa2b3d26cb Fix typos (#107) 
Signed-off-by: micaela <micaelaverucchi@gmail.com>
 2020-09-11 09:13:59 +02:00
Micaela Verucchi fe2e4eae92 yolo4tiny works on tensorRT 🐬 🐬 🐬 🐬 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-06-30 15:52:58 +02:00
Francesco Gatti cbfc8ea4f2 serialize fix 2020-06-14 13:01:45 +02:00
Francesco Gatti 62e4a3f779 memory release 2020-06-02 12:43:06 +02:00
Francesco Gatti d2e2669b6d darknet parse all net to be tested 2020-06-01 12:22:55 +02:00
Micaela Verucchi db0f8a4d99 Support yolov4 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-04-28 20:35:30 +02:00
Francesco Gatti 3d940a9fa2 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
Francesco Gatti 7c81c5a43c batch size > 1 2020-04-21 19:20:44 +02:00
Micaela Verucchi 2fbac7705d Move download of weights inside build folder 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-04-17 22:36:25 +02:00
Micaela Verucchi 3502c5b676 Fix pooling, add return code in each test, add return code handling in test_all_tests script 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-04-09 15:25:05 +02:00
Micaela Verucchi 6d1fda0c21 Compile on tx2 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-04-08 16:23:08 +02:00
Davide Sapienza 26dcb7b8c9 Fix the tensorRT code to support versions prior to 6.0 
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-04-06 19:54:40 +02:00
Davide Sapienza e3442e4764 Move the calibration table saving to the tests/network folder. 
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-04-06 18:06:21 +02:00
Davide Sapienza bb157be82c Add TKDNN_CALIB_IMG_PATH and TKDNN_CALIB_LABRL_PATH variable 
This commit adds two variables for the calibration dataset.
The first is reffered to .txt file that contains the list
of the absolute paths of the images for the INT8 calitration.
The second is referred to .txt file that contains the list
of the absolute paths of the labels of the same images above.

Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-03-30 18:55:32 +02:00
Davide Sapienza 08e1801c60 Add the Int8 calibrator and the tensorRT Int8 inference 
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-03-27 00:48:14 +01:00
Micaela Verucchi 43fd92f701 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
Francesco Gatti 555b32b5dc Refactoring 
Signed-off-by: Francesco Gatti <gattifrancesco@hotmail.it>
 2020-03-14 17:22:58 +01:00
Francesco Gatti 13b4dffd71 csresnext50-panet-spp works with TensorRT 
Signed-off-by: Francesco Gatti <gattifrancesco@hotmail.it>
 2020-03-12 18:24:41 +01:00
xavier 3eb079dd13 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
xavier 38a1b9dcb2 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
Davide Sapienza 2503eba173 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
Davide Sapienza 4616be0738 Add grouped convolutions in CUDNN and tensorRT. 
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-02-05 14:32:58 +01:00
Davide Sapienza 7f239efdc0 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
Davide Sapienza c68d6f318e Merge branch 'master' of https://github.com/ceccocats/tkDNN into cnet 2020-01-20 16:41:19 +01:00
luca 8a4d1cac17 compile with tensorrt 5 2020-01-20 14:51:38 +01:00
Davide Sapienza 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
xavier da4f246157 add DLA, plugin for shortcut and leaky. new verison 0.4 2020-01-15 21:48:18 +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
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 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
Davide Sapienza 02936fa928 ResNet working 2019-10-29 17:21:31 +01:00