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

207 Commits

Author SHA1 Message Date
Francesco Gatti cbfc8ea4f2 serialize fix 2020-06-14 13:01:45 +02:00
Francesco Gatti 20303ac32e cudnn8 compile 2020-06-09 20:53:05 +02:00
Francesco Gatti 62e4a3f779 memory release 2020-06-02 12:43:06 +02:00
Francesco Gatti 0458f361b1 release layer wgs and version update 2020-06-01 21:03:37 +02:00
Francesco Gatti d4e0d07e09 merge 2020-06-01 15:34:59 +02:00
Francesco Gatti 18794d52c2 darknet parser tested 2020-06-01 14:55:08 +02:00
Francesco Gatti d2e2669b6d darknet parse all net to be tested 2020-06-01 12:22:55 +02:00
Micaela Verucchi d936e5f740 Add mish_yashas 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-05-30 16:57:53 +02:00
Micaela Verucchi 854e1d909a Fix original size 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-05-19 16:33:45 +02:00
Micaela Verucchi 5d01a3f629 Fix Centernet postprocessing 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-05-14 17:13:38 +02:00
Micaela Verucchi 40456592fc 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
Micaela Verucchi f684a2126e Merge branch 'master' of https://github.com/ceccocats/tkDNN into eval 2020-05-11 11:58:51 +02:00
Micaela Verucchi 533bb48789 Add json detection creation for codalab check 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-05-11 11:57:58 +02:00
Francesco Gatti 85bbbf42a1 merge 2020-05-03 15:58:00 +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 e6d435a463 Merge commit '04bd5d7ff46270c732b76cd9656e22de930e138e' into tree 2020-04-28 14:54:18 +02:00
Micaela Verucchi a874fad2bd Merge branch 'master' of https://github.com/ceccocats/tkDNN into cnet 2020-04-28 14:43:24 +02:00
dsapienza 6e5ab031c4 Deformable batch works 
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-04-25 02:47:31 +02:00
Micaela Verucchi f418a1f9b2 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
Francesco Gatti 3ba276a236 merge tkDNN 2020-04-24 11:18:22 +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
Francesco Gatti 4a308467e6 warning fix 2020-04-14 14:26:06 +02:00
Francesco Gatti 85588f7343 warning fix 2020-04-09 20:05:10 +02:00
Davide Sapienza 4361d5fec0 Remove the final parameter from the layers 
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-04-09 18:48:03 +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
Francesco Gatti ed0596a52c warning fix 2020-04-09 01:00:14 +02:00
Micaela Verucchi 6d1fda0c21 Compile on tx2 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-04-08 16:23:08 +02:00
Micaela Verucchi 0951f4644f 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
Micaela Verucchi ef78d7f176 Add BoundingBox class 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-04-08 13:52:29 +02:00
Micaela Verucchi 5f6011b206 Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet 2020-04-08 11:20:56 +02:00
Micaela Verucchi ae1d8cd9e6 Refactoring & documentation 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-04-08 11:20:48 +02:00
Davide Sapienza 0e5c90634f Fix the Deformable convolution code sintax. 
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-04-08 11:14:34 +02:00
Davide Sapienza 5562f599a6 Fix the INT8 calibrator sintax 
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-04-08 10:22:03 +02:00
Micaela Verucchi 326c7e0940 Move extraction of name into function in utils 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-04-08 10:09:15 +02:00
Davide Sapienza 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
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
Micaela Verucchi 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
Micaela Verucchi 9ec0913d7f Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet 2020-04-06 18:13:02 +02:00
Micaela Verucchi 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
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
Micaela Verucchi 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
Davide Sapienza e4900120db Fix getNetworkRTName (works also on Xavier) 
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-04-02 10:15:30 +02:00
Micaela Verucchi 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
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 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
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 3eda9b9219 Add GPU version for yolo3detection 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-03-23 21:29:29 +01:00
Micaela Verucchi 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