Francesco Gatti
0621ed223b
Merge branch 'darknetparser' of https://github.com/ceccocats/tkDNN into darknetparser
2020-05-30 19:38:31 +02:00
Francesco Gatti
0ae96a6bc4
darknet parser to be tested on yolo3
2020-05-30 19:38:26 +02:00
Micaela Verucchi
d45947eec5
Add group field
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-05-30 19:03:29 +02:00
Micaela Verucchi
e5e6654b1d
Add fields to darknetParseFields
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-05-30 18:44:17 +02:00
Francesco Gatti
298487d5c5
parse yolo layers
2020-05-30 18:44:02 +02:00
Micaela Verucchi
6b6dbabba3
Modify darknetFields_t
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-05-30 18:10:14 +02:00
Francesco Gatti
e1d5f58c3f
layer parser
2020-05-30 18:02:20 +02:00
Micaela Verucchi
041a5cf65d
Add some fields
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-05-30 18:00:22 +02:00
Francesco Gatti
8935d85e91
parse layer and network
2020-05-30 17:59:00 +02:00
Micaela Verucchi
64d22c51f1
Add darknetParseFields
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-05-30 17:57:48 +02:00
Micaela Verucchi
eec8de3efa
merge
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-05-30 17:33:18 +02:00
Francesco Gatti
548a3dd33c
parse line by line
2020-05-30 17:27:22 +02:00
Micaela Verucchi
15105e90d3
Add parseType
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-05-30 17:25:48 +02:00
Francesco Gatti
4e1c7a70b1
darknet parser interface
2020-05-30 16:58:40 +02:00
Francesco Gatti
ca7631250c
filter yaw ok
2020-05-03 19:12:10 +02:00
Francesco Gatti
7f0d9a930d
Merge branch 'tree' into ipmslam
2020-05-03 15:58:43 +02:00
Francesco Gatti
5a9ed44b6a
SEP model
2020-05-03 15:54:39 +02:00
Francesco Gatti
98537624cf
merge tree
2020-05-01 15:39: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
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
Francesco Gatti
c975a467b1
sensor close if not started fix
2020-04-28 12:59:30 +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
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
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
85588f7343
warning fix
2020-04-09 20:05:10 +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
c1247930c9
gnuplot imuodom
2020-04-08 16:53:35 +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
8575059666
Add again evalutation.h
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-06 18:35:52 +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