cf3fbeddbd
Fix the sorting kernels used in the CenterNet pre and post-processing.
...
This commit moves the kernels in the correct sub-directory.
It creates new header file for Thrust kernels. It splits the
kernels into two files: 'normalize.cu' contains CenterNet
pre-processing operations, 'postprocessing.cu' contains the
CenterNet post-processing operations.
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-04-07 18:28:12 +02:00
ae876e22ee
Update README.md for INT8 inference
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-04-06 20:27:59 +02:00
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
455291b6dd
Fix compile errors dut to Int8BatchStream
...
Signed-off-by: Micaela Verucchi <micaela.verucchi@unimore.it >
Davide Sapienza <sapienza.dav@gmail.com >
2020-04-06 19:11:39 +02:00
8575059666
Add again evalutation.h
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-06 18:35:52 +02:00
9ec0913d7f
Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet
2020-04-06 18:13:02 +02:00
c8308963df
Add evaluation to tk::dnn namespace, style fix also
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-06 18:12:38 +02:00
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
c0a978a480
Add opencv install script (for real)
...
Signed-off-by: Micaela Verucchi <micaela.verucchi@unimore.it >
2020-04-06 17:27:25 +02:00
674c61d280
Add opencv script, modified macro, add opencv section into readme
...
Signed-off-by: Micaela Verucchi <micaela.verucchi@unimore.it >
2020-04-06 17:25:53 +02:00
dfdeb8b36e
Modify weights download for some nets, now also calibration tables are downloaded in tests folder
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-06 16:22:23 +02:00
e4900120db
Fix getNetworkRTName (works also on Xavier)
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-04-02 10:15:30 +02:00
79a05a8159
Modify weights download link for mobilenetv2ssd512 and yolo3_tiny512
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-02 09:41:24 +02:00
af13e7c954
Add getMemoryUsage function, detection update moved in abstract lass, splitted execution time in pre-inf-post, other minors.
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-01 19:18:43 +02:00
c06c9fcf23
Update README.md
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-03-30 19:02:55 +02:00
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
ce20868bad
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
573924cf2c
Update README.md
2020-03-29 20:15:15 +02:00
f2a4125d9f
Update README.md
2020-03-29 20:07:27 +02:00
d7276c720d
Update README.md
2020-03-29 20:03:59 +02:00
482e122655
Fix opencv cuda include error
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-03-27 10:43:49 +01:00
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
7c55dcb708
Add tp tests for yolo3512 and yolo3tiny512
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-03-26 15:06:49 +01:00
3eda9b9219
Add GPU version for yolo3detection
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-03-23 21:29:29 +01:00
df37e11709
Modify map demo, using abstract class. Move draw function in abstract class
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-03-23 19:32:37 +01:00
bbcc33c0cf
Refactoring for detection NN
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-03-20 21:14:12 +01:00
c7d9c38ea0
Add OPENCV_CUDA define, to allow having preprocess both in CPU and GPU (Mobilenet and Centernet)
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-03-20 10:21:11 +01:00
58fe723b6a
Style fix, useless params removed
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-03-19 18:13:35 +01:00
f44f377771
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
41b1135fb3
Add support for mobilenet in map_demo
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-03-17 15:08:43 +01:00
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
1d388ca51a
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
9c25d15ff2
Add yolov3_512 (size 512) test
2020-03-16 11:29:15 +01:00
82d907971c
Fix download weights yolov3 versions
...
Signed-off-by: Micaela Verucchi <micaela.verucchi@unimore.it >
2020-03-14 18:20:32 +01:00
555b32b5dc
Refactoring
...
Signed-off-by: Francesco Gatti <gattifrancesco@hotmail.it >
2020-03-14 17:22:58 +01:00
cf7fbadcd2
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
5e0c1879bc
Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet
2020-03-13 12:17:21 +01:00
32dd807ab0
Add Yolov3_tiny512 test cpp
...
Signed-off-by: xavier <micaelaverucchi@gmail.com >
2020-03-13 12:16:18 +01:00
8aa792925d
Update README.md
2020-03-12 20:42:44 +01:00
9d17cb42f4
Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet
2020-03-12 20:18:11 +01:00
db10ebeb34
Fix print on times.csv
...
Signed-off-by: xavier <micaelaverucchi@gmail.com >
2020-03-12 20:18:02 +01:00
62e7882744
Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet
2020-03-12 18:43:33 +01:00
13b4dffd71
csresnext50-panet-spp works with TensorRT
...
Signed-off-by: Francesco Gatti <gattifrancesco@hotmail.it >
2020-03-12 18:24:41 +01:00
f6527f51e3
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
66e3b98b04
Update README.md
2020-03-11 12:04:28 +01:00
0cc2666e26
Update README.md
2020-03-11 11:51:28 +01:00
d0d462c015
Merge
...
Signed-off-by: xavier <micaelaverucchi@gmail.com >
2020-03-10 15:43:26 +01:00
2d46f16b70
Add tiny yolo 512 test
...
Signed-off-by: xavier <micaelaverucchi@gmail.com >
2020-03-10 15:40:11 +01:00
e6aa73d7bc
Update README.md
2020-03-10 15:26:57 +01:00
4b85a2238a
Add writing results on file for map demo
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-03-10 14:35:25 +01:00
Davide Sapienza
Micaela Verucchi
Micaela Verucchi
Francesco Gatti