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
perseusdg
Micaela Verucchi
Francesco Gatti
Omar Alvarez
Micaela Verucchi