Micaela Verucchi
5eb0cbfcef
Add json detection creation for codalab check
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-05-11 11:57:58 +02:00
Micaela Verucchi
3be4fe0117
Modify download
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-28 20:40:35 +02:00
Micaela Verucchi
1f4f27d752
Support yolov4
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-28 20:35:30 +02:00
Micaela Verucchi
fdfda38fbf
Merge branch 'master' of https://github.com/ceccocats/tkDNN into cnet
2020-04-28 14:43:24 +02:00
Francesco Gatti
f91fb7f287
stream
2020-04-28 14:36:02 +02:00
Micaela Verucchi
dbe54f4b67
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
26d8d009c5
Deformable batch works
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-04-25 02:47:31 +02:00
Micaela Verucchi
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
Francesco Gatti
6a185564e7
RegionRT softmax fix
2020-04-24 00:08:06 +02:00
Francesco Gatti
afaeb75d84
Reshape batch fix
2020-04-23 23:51:23 +02:00
Francesco Gatti
2a6f7db4af
Flatten batch to be checked
2020-04-23 01:18:27 +02:00
Francesco Gatti
1cbf18c21e
test batch
2020-04-23 00:54:03 +02:00
Francesco Gatti
9655c29b94
RouteRt is not used
2020-04-21 19:50:45 +00:00
Francesco Gatti
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
Francesco Gatti
ef91439c87
batch size > 1
2020-04-21 19:20:44 +02:00
Francesco Gatti
ecb2f5e3ad
Merge pull request #2 from ceccocats/cnet
...
Move download of weights inside build folder
2020-04-18 12:43:23 +02:00
Micaela Verucchi
ec7620ebb6
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
4cd1f544f7
Create LICENSE
2020-04-09 22:13:50 +02:00
Francesco Gatti
cc31f78ef2
readme update
2020-04-09 20:15:37 +02:00
Francesco Gatti
585f66e2f6
warning fix
2020-04-09 20:05:10 +02:00
Francesco Gatti
fc873c5219
cmake dont download test data
2020-04-09 20:00:29 +02:00
Francesco Gatti
c71a117eac
ignore calib table
2020-04-09 19:56:08 +02:00
Davide Sapienza
50ada3db60
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
a8fd4cabf3
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
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
Micaela Verucchi
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
Francesco Gatti
c248a94101
warning fix
2020-04-09 01:00:14 +02:00
Francesco Gatti
2389df8626
Merge branch 'master' into cnet
2020-04-09 00:57:43 +02:00
Francesco Gatti
4c5899eae7
remove deprecated and unused warning
2020-04-09 00:55:07 +02:00
Francesco Gatti
73fb12be0f
gnuplot imuodom
2020-04-08 16:53:35 +02:00
Micaela Verucchi
18aff44c99
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
ddae9fe59f
Compile on tx2
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-08 16:23:08 +02:00
Micaela Verucchi
01e667d4f9
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
56c03e3a12
Minor, uncomment test_all_tests
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-08 15:05:52 +02:00
Micaela Verucchi
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
Micaela Verucchi
4d1c0f632e
Add scripts folder
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-08 14:24:34 +02:00
Micaela Verucchi
51ec90e601
Ok test script, output removed
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-08 14:06:30 +02:00
Micaela Verucchi
34968eb4ad
Add BoundingBox class
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-08 13:52:29 +02:00
Micaela Verucchi
3bbc3dce89
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
e036c5da34
Update download url
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-08 12:34:02 +02:00
Micaela Verucchi
b070bb3bd8
Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet
2020-04-08 11:20:56 +02:00
Micaela Verucchi
6a08118e3d
Refactoring & documentation
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-08 11:20:48 +02:00
Davide Sapienza
5be71cf9b7
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
3a5115578e
Fix the INT8 calibrator sintax
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-04-08 10:22:03 +02:00
Micaela Verucchi
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
Micaela Verucchi
065c4e2b58
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
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
Davide Sapienza
44b3919378
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
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
Micaela Verucchi
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