40aefba507
Add mish_yashas
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-05-30 16:57:53 +02:00
9618d5fc7f
Fix original size
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-05-19 16:33:45 +02:00
b5cc4ec91d
Fix Centernet postprocessing
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-05-14 17:13:38 +02:00
7df266cac4
Adapt detection classes to use batches, adapt demos, update README
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-05-14 16:41:51 +02:00
b8686186e4
Merge branch 'master' of https://github.com/ceccocats/tkDNN into eval
2020-05-11 11:58:51 +02:00
5eb0cbfcef
Add json detection creation for codalab check
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-05-11 11:57:58 +02:00
b34c013ad7
merge
2020-05-03 15:58:00 +02:00
1f4f27d752
Support yolov4
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-28 20:35:30 +02:00
1d284bf7d6
Merge commit '04bd5d7ff46270c732b76cd9656e22de930e138e' into tree
2020-04-28 14:54:18 +02:00
fdfda38fbf
Merge branch 'master' of https://github.com/ceccocats/tkDNN into cnet
2020-04-28 14:43:24 +02:00
26d8d009c5
Deformable batch works
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-04-25 02:47:31 +02:00
a761fef159
Add bdd new tests, add BDD100K_val download, fix yolo3-512 link
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-24 12:39:57 +02:00
7597539076
merge tkDNN
2020-04-24 11:18:22 +02: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
ec7620ebb6
Move download of weights inside build folder
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-17 22:36:25 +02:00
70227b6502
warning fix
2020-04-14 14:26:06 +02:00
585f66e2f6
warning fix
2020-04-09 20:05:10 +02:00
8a575f6447
Remove the final parameter from the layers
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-04-09 18:48:03 +02:00
af761786ea
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
c248a94101
warning fix
2020-04-09 01:00:14 +02:00
ddae9fe59f
Compile on tx2
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-08 16:23:08 +02:00
8c1eb69494
Merge with master works
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
Davide Sapienza <sapienza.dav@gmail.com >
2020-04-08 14:59:42 +02:00
34968eb4ad
Add BoundingBox class
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-08 13:52:29 +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
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
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
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
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
bbef698ea3
Fix getNetworkRTName (works also on Xavier)
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-04-02 10:15:30 +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
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
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
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
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
677b6b2698
Refactoring
...
Signed-off-by: Francesco Gatti <gattifrancesco@hotmail.it >
2020-03-14 17:22:58 +01:00
Micaela Verucchi
Francesco Gatti
dsapienza
Davide Sapienza
Micaela Verucchi