fe07c8fb26
Read classes' names from file
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-06-01 10:34:48 +02:00
4b7e80ae0d
yolo3 parsed ok
2020-05-30 22:11:00 +02:00
99f126320b
Merge branch 'darknetparser' of https://github.com/ceccocats/tkDNN into darknetparser
2020-05-30 19:38:31 +02:00
27a83e79d3
darknet parser to be tested on yolo3
2020-05-30 19:38:26 +02:00
93ffbcf8b2
Add group field
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-05-30 19:03:29 +02:00
3e701e7ba3
Add fields to darknetParseFields
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-05-30 18:44:17 +02:00
664a717353
parse yolo layers
2020-05-30 18:44:02 +02:00
15e4ab4436
Modify darknetFields_t
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-05-30 18:10:14 +02:00
3951ff2582
layer parser
2020-05-30 18:02:20 +02:00
26e6ba52a5
Add some fields
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-05-30 18:00:22 +02:00
1755e734b1
parse layer and network
2020-05-30 17:59:00 +02:00
98173ec15c
Add darknetParseFields
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-05-30 17:57:48 +02:00
8b037781f4
merge
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-05-30 17:33:18 +02:00
179e07e830
parse line by line
2020-05-30 17:27:22 +02:00
cb6dc4f182
Add parseType
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-05-30 17:25:48 +02:00
f9d28c58ca
darknet parser interface
2020-05-30 16:58:40 +02:00
54074780cf
filter yaw ok
2020-05-03 19:12:10 +02:00
3fcb05be64
Merge branch 'tree' into ipmslam
2020-05-03 15:58:43 +02:00
8d45259b15
SEP model
2020-05-03 15:54:39 +02:00
bdace759a9
merge tree
2020-05-01 15:39:00 +02:00
1f4f27d752
Support yolov4
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-28 20:35:30 +02:00
f91fb7f287
stream
2020-04-28 14:36:02 +02:00
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
3aa3097bbb
sensor close if not started fix
2020-04-28 12:59:30 +02:00
26d8d009c5
Deformable batch works
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-04-25 02:47:31 +02:00
6a185564e7
RegionRT softmax fix
2020-04-24 00:08:06 +02:00
afaeb75d84
Reshape batch fix
2020-04-23 23:51:23 +02:00
2a6f7db4af
Flatten batch to be checked
2020-04-23 01:18:27 +02:00
9655c29b94
RouteRt is not used
2020-04-21 19:50:45 +00:00
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
ef91439c87
batch size > 1
2020-04-21 19:20:44 +02:00
585f66e2f6
warning fix
2020-04-09 20:05:10 +02:00
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
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
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
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
c248a94101
warning fix
2020-04-09 01:00:14 +02:00
2389df8626
Merge branch 'master' into cnet
2020-04-09 00:57:43 +02:00
73fb12be0f
gnuplot imuodom
2020-04-08 16:53:35 +02:00
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
34968eb4ad
Add BoundingBox class
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-08 13:52:29 +02:00
b070bb3bd8
Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet
2020-04-08 11:20:56 +02:00
6a08118e3d
Refactoring & documentation
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-08 11:20:48 +02:00
5be71cf9b7
Fix the Deformable convolution code sintax.
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-04-08 11:14:34 +02:00
3a5115578e
Fix the INT8 calibrator sintax
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-04-08 10:22:03 +02:00
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
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
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
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
cddd420942
Add again evalutation.h
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-06 18:35:52 +02:00
Micaela Verucchi
Francesco Gatti
dsapienza
Davide Sapienza
Micaela Verucchi