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

328 Commits

This Branch
This Branch
All Branches
Author SHA1 Message Date
Davide Sapienza 7a677d5c10 Add CenterNet based on Resnet101 for 3D, CUDNN and TensorRT work 
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-05-29 14:54:36 +02:00
Davide Sapienza f07f333ae5 Update README 
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-05-27 18:06:26 +02:00
Davide Sapienza 6bdf47bae6 Add 3D demo program 
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-05-27 18:05:38 +02:00
Davide Sapienza ba8c282384 Add 3D CenterNet detection class 
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-05-27 18:03:09 +02:00
Davide Sapienza 64098ad244 Add CenterNet based on DLA34 for 3D, CUDNN and TensorRT work 
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-05-22 12:51:25 +02:00
Micaela Verucchi adb5a693cd Modify download 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-04-28 20:40:35 +02:00
Micaela Verucchi db0f8a4d99 Support yolov4 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-04-28 20:35:30 +02:00
Micaela Verucchi a874fad2bd Merge branch 'master' of https://github.com/ceccocats/tkDNN into cnet 2020-04-28 14:43:24 +02:00
Francesco Gatti 04bd5d7ff4 stream 2020-04-28 14:36:02 +02:00
Micaela Verucchi 89f91f568e Fix shortcutRT plugin, now works with batches, fix yolo3_512 downlaod link 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-04-28 14:34:12 +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 13c9dc6620 RegionRT softmax fix 2020-04-24 00:08:06 +02:00
Francesco Gatti 1ea6097a97 Reshape batch fix 2020-04-23 23:51:23 +02:00
Francesco Gatti 8cff886ee5 Flatten batch to be checked 2020-04-23 01:18:27 +02:00
Francesco Gatti 488887992c test batch 2020-04-23 00:54:03 +02:00
Francesco Gatti 8caff5f598 RouteRt is not used 2020-04-21 19:50:45 +00: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
Francesco Gatti b1818b81d9 Merge pull request #2 from ceccocats/cnet 
Move download of weights inside build folder
 2020-04-18 12:43:23 +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 f637aa0ea8 Create LICENSE 2020-04-09 22:13:50 +02:00
Francesco Gatti c654610569 readme update 2020-04-09 20:15:37 +02:00
Francesco Gatti 85588f7343 warning fix 2020-04-09 20:05:10 +02:00
Francesco Gatti f9e3f17c0a cmake dont download test data 2020-04-09 20:00:29 +02:00
Francesco Gatti cb5b7a1d98 ignore calib table 2020-04-09 19:56:08 +02:00
Davide Sapienza 106cbb5c73 Remove the optional parameters from the Pooling layer. 
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-04-09 19:20:27 +02:00
Micaela Verucchi fabad7bace Add mnist download, change ERROR_TKDNN into ERROR_TENSORRT 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-04-09 19:02:25 +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
Francesco Gatti 2635d24855 Merge branch 'master' into cnet 2020-04-09 00:57:43 +02:00
Francesco Gatti 1bb9f73a0b remove deprecated and unused warning 2020-04-09 00:55:07 +02:00
Francesco Gatti c1247930c9 gnuplot imuodom 2020-04-08 16:53:35 +02:00
Micaela Verucchi bddb0110ca Add imuodom download and test in script 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-04-08 16:50:40 +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 35e86ca5cb Improve test_all_tests scripts to run in every mode (FP32, FP16, INT8) 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
               Davide Sapienza <sapienza.dav@gmail.com>
 2020-04-08 15:29:49 +02:00
Micaela Verucchi 0debd01ba7 Minor, uncomment test_all_tests 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-04-08 15:05:52 +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 6ec417d89f Add scripts folder 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-04-08 14:24:34 +02:00
Micaela Verucchi ba04328d48 Ok test script, output removed 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-04-08 14:06:30 +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 f089d59d10 Add script to check if every test works fine 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-04-08 13:49:06 +02:00
Micaela Verucchi da79128271 Update download url 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-04-08 12:34:02 +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
Micaela Verucchi c36befaf2b 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