Francesco Gatti
e6d435a463
Merge commit '04bd5d7ff46270c732b76cd9656e22de930e138e' into tree
2020-04-28 14:54:18 +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
Francesco Gatti
3ba276a236
merge tkDNN
2020-04-24 11:18:22 +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
cbea75a8f6
merge master
2020-04-15 14:21:33 +02:00
Francesco Gatti
4a308467e6
warning fix
2020-04-14 14:26:06 +02:00
Francesco Gatti
29b99f4e61
pull from repos
2020-04-14 12:32:50 +02:00
Francesco Gatti
e3e5a133f5
pull from repos
2020-04-12 21:47: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
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
ae876e22ee
Update README.md for INT8 inference
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-04-06 20:27:59 +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