e036c5da34
Update download url
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-08 12:34:02 +02:00
b070bb3bd8
Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet
2020-04-08 11:20:56 +02:00
6a08118e3d
Refactoring & documentation
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-08 11:20:48 +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
3a5115578e
Fix the INT8 calibrator sintax
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-04-08 10:22:03 +02:00
6e55e2376f
Move extraction of name into function in utils
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-08 10:09:15 +02:00
065c4e2b58
Minor fix on output file name in map demo
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-08 09:35:54 +02:00
537b0cfa21
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
44b3919378
Update README.md for INT8 inference
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-04-06 20:27:59 +02:00
52e1a192e3
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
c63ff6ecd2
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
cddd420942
Add again evalutation.h
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-06 18:35:52 +02:00
769284574b
Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet
2020-04-06 18:13:02 +02:00
733755e7e0
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
2fbd24625c
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
bd0ed247e9
Add opencv install script (for real)
...
Signed-off-by: Micaela Verucchi <micaela.verucchi@unimore.it >
2020-04-06 17:27:25 +02:00
92dae7f5c5
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
9a1b673db8
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
bbef698ea3
Fix getNetworkRTName (works also on Xavier)
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-04-02 10:15:30 +02:00
9e08f4125a
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
43567dc3ea
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
e2225d2449
Update README.md
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-03-30 19:02:55 +02:00
03d2b88fa6
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
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
42240de49a
Update README.md
2020-03-29 20:15:15 +02:00
283a07f39c
Update README.md
2020-03-29 20:07:27 +02:00
8d5bd653de
Update README.md
2020-03-29 20:03:59 +02:00
e217a0e83f
Fix opencv cuda include error
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-03-27 10:43:49 +01:00
e540213da6
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
f4b976c793
Add tp tests for yolo3512 and yolo3tiny512
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-03-26 15:06:49 +01:00
785a3d8dd5
Add GPU version for yolo3detection
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-03-23 21:29:29 +01:00
3bc648c07b
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
8b9516da97
Refactoring for detection NN
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-03-20 21:14:12 +01:00
997e857e32
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
5c6d6ae0af
Style fix, useless params removed
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-03-19 18:13:35 +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
f6129311d6
Add support for mobilenet in map_demo
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-03-17 15:08:43 +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
01995abdf8
Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet
2020-03-13 12:17:21 +01:00
51b9780baf
Add Yolov3_tiny512 test cpp
...
Signed-off-by: xavier <micaelaverucchi@gmail.com >
2020-03-13 12:16:18 +01:00
35bbda0ab1
Update README.md
2020-03-12 20:42:44 +01:00
a65c8aa91e
Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet
2020-03-12 20:18:11 +01:00
5c347b1e11
Fix print on times.csv
...
Signed-off-by: xavier <micaelaverucchi@gmail.com >
2020-03-12 20:18:02 +01:00
ef1ebea5e9
Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet
2020-03-12 18:43:33 +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
Micaela Verucchi
Davide Sapienza
Micaela Verucchi
Francesco Gatti