6e555678cf
readme update
2021-06-30 11:39:48 +05:30
047ec10e0d
composeee added
2021-06-25 15:27:52 +05:30
d9f371f32c
fix_bbox
2021-06-25 11:36:32 +05:30
ed41be6358
memory-fix
2021-06-24 20:08:33 +05:30
961e1c236e
docker_api
2021-06-11 06:24:36 +00:00
a672254faa
dockerapi
2021-06-09 08:34:35 +00:00
097c38a109
dockerapi
2021-06-09 08:27:32 +00:00
bd0c692628
first docker
2021-06-04 05:28:58 +00:00
8c2282a6fb
first docker
2021-06-04 05:24:13 +00:00
8cc5ebf832
test1
2021-06-03 10:34:26 +05:00
50cb2ca253
fix compile error on old tensorrt and remove useless prints
2021-05-16 22:33:45 +02:00
c2fec5cc8f
remove wrong cuda arch that cause big performance gap
...
fix #226
2021-05-16 22:16:21 +02:00
471d4b27d0
Merge pull request #199 from rickymedrano/rickymedrano-patch-1
...
Minor Readme Changes
2021-04-24 19:00:58 +02:00
3d0742ad54
Merge branch 'perseusdg-master'
2021-04-24 18:55:44 +02:00
1a5d68ae39
merge perseusdg windows10 support
2021-04-24 18:55:26 +02:00
b006e16649
Update README.md
2021-04-14 17:22:26 +05:30
01fa9b668b
Update CMakeLists.txt
2021-04-09 13:20:44 +05:30
2b7ebb8262
Code cleanup and readme fixes
2021-04-09 13:17:41 +05:30
305cdd8b12
minor modifications
2021-04-08 00:57:58 +05:30
d00f051768
merge from gitlab
2021-04-08 00:28:00 +05:30
29b0c551a5
Opencv cuda fix
2021-03-31 03:44:51 -07:00
516017cc8f
Update download_validation.py
2021-03-26 15:09:28 +01:00
06b7538da1
python download file
2021-03-26 19:37:44 +05:30
a2b0dd62d1
added yolo4x and yolo4-csp from the github repo and download file corrections
2021-03-26 12:18:46 +05:30
557fb732a5
ReadMe.md windows changes
2021-03-25 20:46:12 +05:30
3b2df9ee85
ReadMe.md windows changes
2021-03-25 20:44:43 +05:30
7620ee48e7
timer fix
2021-03-25 16:38:53 +05:30
a742245116
tkdnn first patch for windows
2021-03-24 22:26:46 +05:30
a1c800632d
minor migrations
2021-03-17 23:06:50 +05:30
f52faeefe4
shared_ptr migrations
2021-03-17 22:26:26 +05:30
0180f7830b
Commits for msvc 16.9
2021-03-10 12:59:04 +05:30
bca76f7221
Minor Readme Changes
...
Added Logistic as a viable activation you can use.
Added conf-thresh as the 7th parameter in the ./demo call.
2021-02-09 07:57:57 -08:00
5b0a1bb4cd
minor fixes in test.h
2021-01-26 23:11:58 +05:30
c935e8190e
Builds on windows successfully,issues with deserialization and downloading weights
2021-01-26 20:17:43 +04:00
5995048f63
Replaced dynamic arrays with std::vector ,works on linux ..needs to be tested on windows after clearing up the lnk2019 error
2021-01-25 04:12:05 +05:30
703bc2d660
Add support to Scaled-YOLO4, update Yolov4x-mish (tested)
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2021-01-22 17:57:42 +01:00
6327e46513
minor fixes
2021-01-21 09:29:27 +04:00
b994eb2a4e
able to build kernels as shared object file(dll),and minor changes to lstm.cpp and utils.cpp to overcome minor msvc build errors
2021-01-21 00:22:15 +04:00
4fa008bd85
Update README
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-11-23 11:34:06 +01:00
936068e9b4
Add support for yolov4x-mish.
...
Changes:
- add parameters nms_kind, nms_thresh, new_coords to yolo layer and darknet parser
- added diou nms, new method to compute the BBs
- created test for yolov4x-mish called yolo4x
Tested, all tests work.
Problem to solve: little loss in mAP of yolo4x
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-11-23 11:29:32 +01:00
4f3cdd76c8
Add yolo4_mmr test
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-10-23 11:40:55 +02:00
08f0d3eb34
Update README.md
2020-10-10 13:03:01 +02:00
b10e446a11
Update README with Xavier NX FPS results 15W4Core
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-09-15 14:09:02 +02:00
6bafa0007f
Update README with Xavier NX FPS results
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-09-15 10:50:24 +02:00
a194ba59f6
Fix typos ( #107 )
...
Signed-off-by: micaela <micaelaverucchi@gmail.com >
2020-09-11 09:13:59 +02:00
6d1f6b3881
Merge branch 'master' of https://github.com/ceccocats/tkDNN
2020-08-06 15:54:09 +02:00
c7be331acd
Fix boxes also for Centernet
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-08-06 15:53:57 +02:00
096a67281b
Update README.md
2020-08-06 11:05:26 +02:00
498b163f36
Fixed boxes to float, add conf thresh as param
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-08-05 19:55:10 +02:00
8c8341cc61
fix coords convert
2020-07-27 13:52:39 +02:00
b4015a8e26
Resolve detection objects pick by prob threshold.
...
Before this it will only pick the last object with prob > thresh wich is absolutely wrong
Now it picks all the objects with prob > thesh.
fixes #94
2020-07-27 13:45:42 +02:00
e5991378d5
Update README.md
2020-07-17 14:37:10 +02:00
921ed60cf5
Fix patch from @ahmedius2 , tkDNN now supports CUDNN 8.0.1 ( Fix #74 )
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
Francesco Gatti <gattifrancesco@hotmail.it >
2020-07-16 19:06:54 +02:00
5c7f1c283e
Patch for CUDNN 8.0.1
...
Signed-off-by: tk <micaelaverucchi@gmail.com >
2020-07-16 18:16:09 +02:00
4959f4fa9e
Update README.md
2020-07-13 19:51:09 +02:00
79bd3d6f35
docker
2020-07-13 19:50:00 +02:00
83d57436ac
Add weights download link for csresnext50-panet-spp_berkeley ( fix #63 )
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-07-01 11:56:46 +02:00
8fad424c61
Yolo4-tiny batched
...
fix #59
2020-06-30 19:37:16 +02:00
4c9b9bf03a
yolo4tiny works on tensorRT 🐬 🐬 🐬 🐬
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-06-30 15:52:58 +02:00
35c0d9fd37
yolov4tiny works on CUDNN
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-06-30 15:19:03 +02:00
562ba06cd8
Merge pull request #53 from omaralvarez/master
...
Fix error when parsing label files with unexpected chars
2020-06-25 10:07:41 +02:00
98ecd94494
Fix parsing label files with unexpected chars
2020-06-22 13:40:15 +02:00
6d467e5d93
viz yolo3 preprocess
2020-06-16 13:19:30 +02:00
517eaa553b
dealloc in test.h
...
fix #36
2020-06-16 12:44:12 +02:00
f9c50b2a49
darknet parser cpp
2020-06-16 12:41:48 +02:00
0f3dbac962
viz yolo3
2020-06-16 12:36:45 +02:00
1e8ef497d8
Imptove relative paths
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
Francesco Gatti <gattifrancesco@hotmail.it >
2020-06-15 17:15:37 +02:00
fec43f35dc
fix wrong commit
2020-06-14 13:03:48 +02:00
d617062752
serialize fix
2020-06-14 13:01:45 +02:00
ffe332ebb1
cmake cudnn fix
2020-06-14 12:57:29 +02:00
971de08e9a
Update README
...
Signed-off-by: Micaela Verucchi <micaela.verucchi@unimore.it >
2020-06-12 11:39:46 +02:00
863e27a04e
Fix minor, update readme
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-06-09 21:01:19 +02:00
9d19370bbb
Merge branch 'master' of https://github.com/ceccocats/tkDNN
2020-06-09 20:53:17 +02:00
b419fdfc04
cudnn8 compile
2020-06-09 20:53:05 +02:00
95c240223e
Update README.md
2020-06-03 11:01:17 +02:00
4f907dde40
memory release
2020-06-02 12:43:06 +02:00
8c79b75227
release layer wgs and version update
2020-06-01 21:03:37 +02:00
8e7a355fab
readme update
2020-06-01 19:04:39 +02:00
235a6838eb
all test ok
2020-06-01 16:15:29 +02:00
2df8bb92c2
merge
2020-06-01 15:34:59 +02:00
7259511ea2
Merge branch 'master' of https://github.com/ceccocats/tkDNN into eval
2020-06-01 15:18:29 +02:00
fd00822a3e
Modify yolov4_berkeley link for download, clean to merge with master
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-06-01 15:17:53 +02:00
af9bb411be
darknet parser tested
2020-06-01 14:55:08 +02:00
21518c5eef
darknet parse all net to be tested
2020-06-01 12:22:55 +02:00
702e38a150
Add names files
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-06-01 10:42:06 +02:00
9ee8c5eb58
Check
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-06-01 10:39:13 +02:00
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
40aefba507
Add mish_yashas
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-05-30 16:57:53 +02:00
0c3f4b054b
Update LICENSE
2020-05-28 14:44:07 +02:00
9618d5fc7f
Fix original size
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-05-19 16:33:45 +02:00
f2991bec5c
Add variable batch for detector update
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-05-19 09:32:27 +02:00
ad21cc0b79
Add yolo4_berkeley test
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-05-15 09:48:59 +02:00
360adbd1a1
Add verbose define
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-05-15 09:41:27 +02:00
b5cc4ec91d
Fix Centernet postprocessing
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-05-14 17:13:38 +02:00
7df266cac4
Adapt detection classes to use batches, adapt demos, update README
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-05-14 16:41:51 +02:00
7e8dbc8296
Add flag to disable visualization and save video
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-05-11 15:08:10 +02:00
b8686186e4
Merge branch 'master' of https://github.com/ceccocats/tkDNN into eval
2020-05-11 11:58:51 +02:00
5eb0cbfcef
Add json detection creation for codalab check
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-05-11 11:57:58 +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
b34c013ad7
merge
2020-05-03 15:58:00 +02:00
099aa1accd
Imu odom weights
2020-05-03 15:56:30 +02:00
eb1d30d1b1
Merge branch 'master' of https://github.com/ceccocats/tkDNN
2020-05-03 15:54:42 +02:00
8d45259b15
SEP model
2020-05-03 15:54:39 +02:00
e3b60f0288
Merge branch 'master' of https://github.com/ceccocats/tkDNN
2020-05-03 15:53:51 +02:00
72dfd92c0a
imu odom SEP model
2020-05-03 15:53:35 +02:00
bdace759a9
merge tree
2020-05-01 15:39:00 +02:00
0ef8aaf616
Update README.md
2020-04-29 18:34:06 +02:00
6f51c1262c
Update README.md
2020-04-29 18:17:32 +02:00
5ac29ebb12
Update README.md
2020-04-29 18:09:50 +02:00
e766df9f25
Update README.md
2020-04-29 13:07:26 +02:00
a898825af8
Update LICENSE
2020-04-28 22:59:45 +02:00
3be4fe0117
Modify download
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-28 20:40:35 +02:00
1f4f27d752
Support yolov4
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-28 20:35:30 +02:00
1d284bf7d6
Merge commit '04bd5d7ff46270c732b76cd9656e22de930e138e' into tree
2020-04-28 14:54:18 +02:00
fdfda38fbf
Merge branch 'master' of https://github.com/ceccocats/tkDNN into cnet
2020-04-28 14:43:24 +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
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
7597539076
merge tkDNN
2020-04-24 11:18:22 +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
1cbf18c21e
test batch
2020-04-23 00:54:03 +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
ecb2f5e3ad
Merge pull request #2 from ceccocats/cnet
...
Move download of weights inside build folder
2020-04-18 12:43:23 +02:00
ec7620ebb6
Move download of weights inside build folder
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-17 22:36:25 +02:00
431819a644
merge master
2020-04-15 14:21:33 +02:00
70227b6502
warning fix
2020-04-14 14:26:06 +02:00
2df8f87cb8
pull from repos
2020-04-14 12:32:50 +02:00
b9a354b13a
pull from repos
2020-04-12 21:47:25 +02:00
4cd1f544f7
Create LICENSE
2020-04-09 22:13:50 +02:00
cc31f78ef2
readme update
2020-04-09 20:15:37 +02:00
585f66e2f6
warning fix
2020-04-09 20:05:10 +02:00
fc873c5219
cmake dont download test data
2020-04-09 20:00:29 +02:00
c71a117eac
ignore calib table
2020-04-09 19:56:08 +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
4c5899eae7
remove deprecated and unused warning
2020-04-09 00:55:07 +02:00
73fb12be0f
gnuplot imuodom
2020-04-08 16:53:35 +02:00
18aff44c99
Add imuodom download and test in script
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-08 16:50:40 +02:00
ddae9fe59f
Compile on tx2
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-08 16:23:08 +02:00
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
56c03e3a12
Minor, uncomment test_all_tests
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-08 15:05:52 +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
4d1c0f632e
Add scripts folder
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-08 14:24:34 +02:00
51ec90e601
Ok test script, output removed
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-08 14:06:30 +02:00
34968eb4ad
Add BoundingBox class
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-08 13:52:29 +02:00
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
e036c5da34
Update download url
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-08 12:34:02 +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
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
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
44b3919378
Update README.md for INT8 inference
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-04-06 20:27:59 +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
769284574b
Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet
2020-04-06 18:13:02 +02:00
733755e7e0
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
2fbd24625c
Move the calibration table saving to the tests/network folder.
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-04-06 18:06:21 +02:00
bd0ed247e9
Add opencv install script (for real)
...
Signed-off-by: Micaela Verucchi <micaela.verucchi@unimore.it >
2020-04-06 17:27:25 +02:00
92dae7f5c5
Add opencv script, modified macro, add opencv section into readme
...
Signed-off-by: Micaela Verucchi <micaela.verucchi@unimore.it >
2020-04-06 17:25:53 +02:00
9a1b673db8
Modify weights download for some nets, now also calibration tables are downloaded in tests folder
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-06 16:22:23 +02:00
bbef698ea3
Fix getNetworkRTName (works also on Xavier)
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-04-02 10:15:30 +02:00
9e08f4125a
Modify weights download link for mobilenetv2ssd512 and yolo3_tiny512
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-02 09:41:24 +02:00
43567dc3ea
Add getMemoryUsage function, detection update moved in abstract lass, splitted execution time in pre-inf-post, other minors.
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-01 19:18:43 +02:00
e2225d2449
Update README.md
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-03-30 19:02:55 +02:00
03d2b88fa6
Add TKDNN_CALIB_IMG_PATH and TKDNN_CALIB_LABRL_PATH variable
...
This commit adds two variables for the calibration dataset.
The first is reffered to .txt file that contains the list
of the absolute paths of the images for the INT8 calitration.
The second is referred to .txt file that contains the list
of the absolute paths of the labels of the same images above.
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-03-30 18:55:32 +02:00
0474f019cf
Add TKDNN_MODE variable to the name of the network.
...
This commit permits to obtain different .rt files for different
precision optimizations of the same network.
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-03-30 15:08:25 +02:00
42240de49a
Update README.md
2020-03-29 20:15:15 +02:00
283a07f39c
Update README.md
2020-03-29 20:07:27 +02:00
8d5bd653de
Update README.md
2020-03-29 20:03:59 +02:00
e217a0e83f
Fix opencv cuda include error
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-03-27 10:43:49 +01:00
e540213da6
Add the Int8 calibrator and the tensorRT Int8 inference
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-03-27 00:48:14 +01:00
f4b976c793
Add tp tests for yolo3512 and yolo3tiny512
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-03-26 15:06:49 +01:00
785a3d8dd5
Add GPU version for yolo3detection
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-03-23 21:29:29 +01:00
3bc648c07b
Modify map demo, using abstract class. Move draw function in abstract class
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-03-23 19:32:37 +01:00
8b9516da97
Refactoring for detection NN
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-03-20 21:14:12 +01:00
997e857e32
Add OPENCV_CUDA define, to allow having preprocess both in CPU and GPU (Mobilenet and Centernet)
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-03-20 10:21:11 +01:00
5c6d6ae0af
Style fix, useless params removed
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-03-19 18:13:35 +01:00
cae0b84687
Add preprocess function, allow preprocess on GPU for mobilenetdetection
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-03-19 17:53:29 +01:00
f6129311d6
Add support for mobilenet in map_demo
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-03-17 15:08:43 +01:00
1a50065bde
Fix conv2d with additional bias for tensorRT. Fix reshape deserialize. Mb2512 works with tensorRT
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-03-17 13:18:05 +01:00
14d850751d
Add mobilenetv2ssdlite512 test.
...
Works for CuDNN, not for tensorRT.
Modified channels in second convolution for classification headers from 126 to 486, when changing size from 300 to 512.
Added size 512 SSD specs and support to COCO dataset (81 classes, first BACKGROUND due to repo for training).
Refactoring class names.
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-03-17 10:35:02 +01:00
4a2a244162
Add yolov3_512 (size 512) test
2020-03-16 11:29:15 +01:00
a7c3761960
eigen assert fix
2020-03-14 19:50:55 +01:00
94971fbdcb
Fix download weights yolov3 versions
...
Signed-off-by: Micaela Verucchi <micaela.verucchi@unimore.it >
2020-03-14 18:20:32 +01:00
677b6b2698
Refactoring
...
Signed-off-by: Francesco Gatti <gattifrancesco@hotmail.it >
2020-03-14 17:22:58 +01:00
21b8146232
Add function to download weigths if do no exist, for each test. Add some controls in map demo.
...
Signed-off-by: Francesco Gatti <gattifrancesco@hotmail.it >
2020-03-14 17:02:07 +01:00
01995abdf8
Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet
2020-03-13 12:17:21 +01:00
51b9780baf
Add Yolov3_tiny512 test cpp
...
Signed-off-by: xavier <micaelaverucchi@gmail.com >
2020-03-13 12:16:18 +01:00
35bbda0ab1
Update README.md
2020-03-12 20:42:44 +01:00
a65c8aa91e
Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet
2020-03-12 20:18:11 +01:00
5c347b1e11
Fix print on times.csv
...
Signed-off-by: xavier <micaelaverucchi@gmail.com >
2020-03-12 20:18:02 +01:00
ef1ebea5e9
Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet
2020-03-12 18:43:33 +01:00
4845edbc88
csresnext50-panet-spp works with TensorRT
...
Signed-off-by: Francesco Gatti <gattifrancesco@hotmail.it >
2020-03-12 18:24:41 +01:00
5f4508a5a6
Add csresnext50-panet-spp test. Works on CUDNN. Does not work with TensorRT
...
Signed-off-by: Francesco Gatti <gattifrancesco@hotmail.it >
2020-03-11 18:37:26 +01:00
963b2c40d1
Update README.md
2020-03-11 12:04:28 +01:00
d5722cae3f
Update README.md
2020-03-11 11:51:28 +01:00
98770ee0c5
Merge
...
Signed-off-by: xavier <micaelaverucchi@gmail.com >
2020-03-10 15:43:26 +01:00
07a61b0e37
Add tiny yolo 512 test
...
Signed-off-by: xavier <micaelaverucchi@gmail.com >
2020-03-10 15:40:11 +01:00
0c5e31476e
Update README.md
2020-03-10 15:26:57 +01:00
75d5e20843
Add writing results on file for map demo
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-03-10 14:35:25 +01:00
2d378e9070
imuodom fix
2020-03-09 22:05:34 +01:00
6fee8d5ec5
- ImuOdom model into class
...
- Route layer input array hard copy
- fix utils
2020-03-09 19:44:21 +01:00
0773f216b3
Update README
...
Signed-off-by: Micaela Verucchi <micaela.verucchi@unimore.it >
2020-03-08 18:12:08 +01:00
b734baca0c
Update README
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-03-03 17:34:01 +01:00
99e809737f
Move multiple demo files into one
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-03-03 17:30:35 +01:00
c94159a2bf
Add draw method to Yolo3Detection class.
...
Update yolo demo
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-03-03 17:08:25 +01:00
1d38d88c85
Fix dependencies problems
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-02-27 16:48:10 +01:00
3be254b08f
Add Mobilenetv2 SSD Lite post and preprocessing, add mobilenet demo
...
Signed-off-by: xavier <micaelaverucchi@gmail.com >
2020-02-26 17:42:20 +01:00
c36c898948
Fix include error
...
Signed-off-by: xavier <micaelaverucchi@gmail.com >
2020-02-21 11:14:52 +01:00
c4cef1aa02
Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet
2020-02-21 10:49:35 +01:00
808a84131c
Add Mobilenet2SSDLite test
...
The new test works both with TensorRT and cuDNN. Preprocessing and
Postprocessing are missing. Add ClippedReLU (for ReLU6), groups for
Conv2d, additional bias for convolution.
Other minors:
-move the timer in the detector to measure all the
processing time for a given frame (both centernet and yolo);
-add int8 flag.
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
Davide Sapienza <sapienza.dav@gmail.com >
2020-02-21 10:45:46 +01:00
cd1e1c1582
LSTM ok
2020-02-16 17:21:58 +01:00
4b48a2f38c
structure ok, result wrong
2020-02-16 17:08:19 +01:00
0ac292ea48
works but it need cleaning
2020-02-16 16:28:39 +01:00
8736a8c3da
LSTM params
2020-02-15 20:37:08 +01:00
200e9466f6
Move pre-processing on GPU
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-02-14 18:44:13 +01:00
f5706d8523
lstm return seq
2020-02-13 23:21:28 +01:00
0329318ca3
removed unused var
2020-02-13 23:10:48 +01:00
e8280d3a65
LSTM to be tested
2020-02-13 23:04:29 +01:00
c876fa05ae
LSTM cudnn test
2020-02-13 19:27:18 +01:00
40a4a55cd0
Fix a bug in the deformable kernel.
...
There was a wrong variable initialization.
Fixes: 51ffcb1f50sapienza.dav@gmail.com >
2020-02-11 16:28:17 +01:00
217b3fbfb5
Update README.md
2020-02-11 15:12:27 +01:00
b1a6b45f5f
Merge README
...
Signed-off-by: xavier <micaelaverucchi@gmail.com >
2020-02-11 14:58:25 +01:00
9c4ea524fc
Add script to download COCO val2017
...
Signed-off-by: xavier <micaelaverucchi@gmail.com >
2020-02-11 14:56:54 +01:00
21535dd6b3
Update README.md
2020-02-11 11:28:20 +01:00
d0026ff235
Change README.md, add config.yaml, refactoring
...
Signed-off-by: xavier <micaelaverucchi@gmail.com >
2020-02-11 11:25:42 +01:00
e01d7c6b13
Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet
2020-02-11 10:32:52 +01:00
73548b179e
Read parameters for mAP from yaml, add yampl-cpp dependency
...
Signed-off-by: xavier <micaelaverucchi@gmail.com >
2020-02-11 10:32:42 +01:00
2b06836a7f
Optimize deformable kernel
...
Signed-off.by: Ignacio Sañudo Olmedo<ignacio.sanudoolmedo@unimore.it >
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-02-11 10:06:08 +01:00
3ef2cc65f5
Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet
2020-02-10 18:19:12 +01:00
c7cb3e7571
Add avg precision, recall, f1score computation, other minors
...
Signed-off-by: xavier <micaelaverucchi@gmail.com >
2020-02-10 18:19:08 +01:00
34ecb6a1bb
Update README.md
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-02-10 11:19:39 +01:00
da8ba7d8ce
Add Anaconda environment for ResNet101 and DLA34 weights exporter
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-02-10 10:58:04 +01:00
932de41e38
Add DLA34 weights exporter
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-02-10 10:56:33 +01:00
5d07e98a9c
Update ResNet101 weights exporter
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-02-10 10:54:04 +01:00
17a083ae28
Remove mallocs and frees from the kernels
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-02-07 14:50:05 +01:00
da24c2e752
Refactoring map_demo, add evaluation.h and evaluation.cpp
...
Signed-off-by: xavier <micaelaverucchi@gmail.com >
2020-02-07 12:47:02 +01:00
9bc4ba0eff
Add mAP 0.5:0.95, other small fix
...
Signed-off-by: xavier <micaelaverucchi@gmail.com >
2020-02-07 09:09:40 +01:00
002355e7a7
Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet
2020-02-07 09:06:29 +01:00
5abf1613f2
Fix memory leak
...
This commit moves cublasCreate out from dcn_v2_cuda_forward
to save some milliseconds and it adds cublasDestroy (cause
of memory leak).
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-02-06 23:09:12 +01:00
0a4c75f834
Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet
2020-02-06 18:24:49 +01:00
516c1e6636
Change CenterNet input dimension.
...
This commit changes the image input dimension, it updates the
CenterNet detection class.
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-02-06 18:08:06 +01:00
dc130fc834
Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet
2020-02-05 18:39:32 +01:00
ef18be7b7b
Add mAP computation and demo
...
Signed-off-by: xavier <micaelaverucchi@gmail.com >
2020-02-05 18:39:13 +01:00
507d30bf52
Implement CenterNet based on DLA34, CUDNN and TensorRT work.
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-02-05 14:59:56 +01:00
0d6acaec41
Add grouped convolutions in CUDNN and tensorRT.
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-02-05 14:32:58 +01:00
04918f539e
Implement DLA34, CUDNN and TensorRT work.
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-01-24 19:30:32 +01:00
07da54efe9
Centernet: fix pooling problem, add centrnet demo
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-01-21 12:50:18 +01:00
b7a6e8c8f5
Merge branch 'master' of https://github.com/ceccocats/tkDNN into cnet
2020-01-20 16:41:19 +01:00
6a3b261bc9
compile with tensorrt 5
2020-01-20 14:51:38 +01:00
c3aadb089b
Update README.md
2020-01-16 18:24:35 +01:00
07eae338fe
Update README.md
2020-01-16 18:21:34 +01:00
da4f246157
add DLA, plugin for shortcut and leaky. new verison 0.4
2020-01-15 21:48:18 +01:00
f3f5daf3db
Merge branch 'master' of https://github.com/ceccocats/tkDNN
2020-01-15 18:07:44 +01:00
c2d73623e5
support clion
2020-01-15 18:07:40 +01:00
mohit
hardik22317
mohit
root
root
root
Francesco Gatti
Harshvardhan Chandirasekar
perseusdg
Harshvardhan Chandirasekar
Ricky Medrano
Micaela Verucchi
Omar Alvarez
Micaela Verucchi
dsapienza
Davide Sapienza
luca