a2b0dd62d1
added yolo4x and yolo4-csp from the github repo and download file corrections
2021-03-26 12:18:46 +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
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
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
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
a194ba59f6
Fix typos ( #107 )
...
Signed-off-by: micaela <micaelaverucchi@gmail.com >
2020-09-11 09:13:59 +02:00
c7be331acd
Fix boxes also for Centernet
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-08-06 15:53:57 +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
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
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
98ecd94494
Fix parsing label files with unexpected chars
2020-06-22 13:40:15 +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
d617062752
serialize fix
2020-06-14 13:01:45 +02:00
b419fdfc04
cudnn8 compile
2020-06-09 20:53:05 +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
2df8bb92c2
merge
2020-06-01 15:34:59 +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
40aefba507
Add mish_yashas
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-05-30 16:57:53 +02:00
9618d5fc7f
Fix original size
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-05-19 16:33:45 +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
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
b34c013ad7
merge
2020-05-03 15:58:00 +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
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
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
ec7620ebb6
Move download of weights inside build folder
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-17 22:36:25 +02:00
70227b6502
warning fix
2020-04-14 14:26:06 +02:00
585f66e2f6
warning fix
2020-04-09 20:05:10 +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
ddae9fe59f
Compile on tx2
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-08 16:23:08 +02:00
hchandirasekar
Harshvardhan Chandirasekar
Micaela Verucchi
Francesco Gatti
Omar Alvarez
dsapienza
Davide Sapienza