f07f333ae5
Update README
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-05-27 18:06:26 +02:00
6bdf47bae6
Add 3D demo program
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-05-27 18:05:38 +02:00
ba8c282384
Add 3D CenterNet detection class
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-05-27 18:03:09 +02:00
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
adb5a693cd
Modify download
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-28 20:40:35 +02:00
db0f8a4d99
Support yolov4
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-28 20:35:30 +02:00
a874fad2bd
Merge branch 'master' of https://github.com/ceccocats/tkDNN into cnet
2020-04-28 14:43:24 +02:00
04bd5d7ff4
stream
2020-04-28 14:36:02 +02:00
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
6e5ab031c4
Deformable batch works
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-04-25 02:47:31 +02:00
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
13c9dc6620
RegionRT softmax fix
2020-04-24 00:08:06 +02:00
1ea6097a97
Reshape batch fix
2020-04-23 23:51:23 +02:00
8cff886ee5
Flatten batch to be checked
2020-04-23 01:18:27 +02:00
488887992c
test batch
2020-04-23 00:54:03 +02:00
8caff5f598
RouteRt is not used
2020-04-21 19:50:45 +00:00
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
7c81c5a43c
batch size > 1
2020-04-21 19:20:44 +02:00
b1818b81d9
Merge pull request #2 from ceccocats/cnet
...
Move download of weights inside build folder
2020-04-18 12:43:23 +02:00
2fbac7705d
Move download of weights inside build folder
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-17 22:36:25 +02:00
f637aa0ea8
Create LICENSE
2020-04-09 22:13:50 +02:00
c654610569
readme update
2020-04-09 20:15:37 +02:00
85588f7343
warning fix
2020-04-09 20:05:10 +02:00
f9e3f17c0a
cmake dont download test data
2020-04-09 20:00:29 +02:00
cb5b7a1d98
ignore calib table
2020-04-09 19:56:08 +02:00
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
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
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
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
ed0596a52c
warning fix
2020-04-09 01:00:14 +02:00
2635d24855
Merge branch 'master' into cnet
2020-04-09 00:57:43 +02:00
1bb9f73a0b
remove deprecated and unused warning
2020-04-09 00:55:07 +02:00
c1247930c9
gnuplot imuodom
2020-04-08 16:53:35 +02:00
bddb0110ca
Add imuodom download and test in script
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-08 16:50:40 +02:00
6d1fda0c21
Compile on tx2
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-08 16:23:08 +02:00
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
0debd01ba7
Minor, uncomment test_all_tests
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-08 15:05:52 +02:00
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
6ec417d89f
Add scripts folder
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-08 14:24:34 +02:00
ba04328d48
Ok test script, output removed
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-08 14:06:30 +02:00
ef78d7f176
Add BoundingBox class
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-08 13:52:29 +02:00
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
da79128271
Update download url
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-08 12:34:02 +02:00
5f6011b206
Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet
2020-04-08 11:20:56 +02:00
ae1d8cd9e6
Refactoring & documentation
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-08 11:20:48 +02:00
0e5c90634f
Fix the Deformable convolution code sintax.
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-04-08 11:14:34 +02:00
5562f599a6
Fix the INT8 calibrator sintax
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-04-08 10:22:03 +02:00
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
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
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
Micaela Verucchi
Francesco Gatti
dsapienza