mmr/tkDNN
Archived
1
0
Fork 0
Code Issues 46 Pull Requests 12 Releases 4 Activity
449 Commits 11 Branches 5 Tags
8c2282a6fb0d411a8d78acae869151f3cb444f9b
Commit Graph

227 Commits

Author SHA1 Message Date
root 8c2282a6fb first docker 2021-06-04 05:24:13 +00:00
Francesco Gatti 50cb2ca253 fix compile error on old tensorrt and remove useless prints 2021-05-16 22:33:45 +02:00
Francesco Gatti 1a5d68ae39 merge perseusdg windows10 support 2021-04-24 18:55:26 +02:00
perseusdg 2b7ebb8262 Code cleanup and readme fixes 2021-04-09 13:17:41 +05:30
perseusdg 305cdd8b12 minor modifications 2021-04-08 00:57:58 +05:30
perseusdg d00f051768 merge from gitlab 2021-04-08 00:28:00 +05:30
hchandirasekar 29b0c551a5 Opencv cuda fix 2021-03-31 03:44:51 -07:00
hchandirasekar a2b0dd62d1 added yolo4x and yolo4-csp from the github repo and download file corrections 2021-03-26 12:18:46 +05:30
hchandirasekar 7620ee48e7 timer fix 2021-03-25 16:38:53 +05:30
hchandirasekar a742245116 tkdnn first patch for windows 2021-03-24 22:26:46 +05:30
Harshvardhan Chandirasekar a1c800632d minor migrations 2021-03-17 23:06:50 +05:30
Harshvardhan Chandirasekar f52faeefe4 shared_ptr migrations 2021-03-17 22:26:26 +05:30
hchandirasekar 5b0a1bb4cd minor fixes in test.h 2021-01-26 23:11:58 +05:30
hchandirasekar c935e8190e Builds on windows successfully,issues with deserialization and downloading weights 2021-01-26 20:17:43 +04:00
Micaela Verucchi 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
perseusdg 6327e46513 minor fixes 2021-01-21 09:29:27 +04:00
perseusdg 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
Micaela Verucchi 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
micaela a194ba59f6 Fix typos (#107) 
Signed-off-by: micaela <micaelaverucchi@gmail.com>
 2020-09-11 09:13:59 +02:00
Micaela Verucchi 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
tk 5c7f1c283e Patch for CUDNN 8.0.1 
Signed-off-by: tk <micaelaverucchi@gmail.com>
 2020-07-16 18:16:09 +02:00
Francesco Gatti 8fad424c61 Yolo4-tiny batched 
fix #59
 2020-06-30 19:37:16 +02:00
Micaela Verucchi 4c9b9bf03a yolo4tiny works on tensorRT 🐬 🐬 🐬 🐬 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-06-30 15:52:58 +02:00
Micaela Verucchi 35c0d9fd37 yolov4tiny works on CUDNN 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-06-30 15:19:03 +02:00
Francesco Gatti 517eaa553b dealloc in test.h 
fix #36
 2020-06-16 12:44:12 +02:00
Francesco Gatti f9c50b2a49 darknet parser cpp 2020-06-16 12:41:48 +02:00
Francesco Gatti 0f3dbac962 viz yolo3 2020-06-16 12:36:45 +02:00
Francesco Gatti 4f907dde40 memory release 2020-06-02 12:43:06 +02:00
Francesco Gatti 8c79b75227 release layer wgs and version update 2020-06-01 21:03:37 +02:00
Francesco Gatti 235a6838eb all test ok 2020-06-01 16:15:29 +02:00
Francesco Gatti 2df8bb92c2 merge 2020-06-01 15:34:59 +02:00
Micaela Verucchi 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
Francesco Gatti af9bb411be darknet parser tested 2020-06-01 14:55:08 +02:00
Francesco Gatti 21518c5eef darknet parse all net to be tested 2020-06-01 12:22:55 +02:00
Micaela Verucchi 9ee8c5eb58 Check 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-06-01 10:39:13 +02:00
Micaela Verucchi fe07c8fb26 Read classes' names from file 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-06-01 10:34:48 +02:00
Francesco Gatti 4b7e80ae0d yolo3 parsed ok 2020-05-30 22:11:00 +02:00
Francesco Gatti 99f126320b Merge branch 'darknetparser' of https://github.com/ceccocats/tkDNN into darknetparser 2020-05-30 19:38:31 +02:00
Francesco Gatti 27a83e79d3 darknet parser to be tested on yolo3 2020-05-30 19:38:26 +02:00
Micaela Verucchi 93ffbcf8b2 Add group field 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-05-30 19:03:29 +02:00
Micaela Verucchi 3e701e7ba3 Add fields to darknetParseFields 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-05-30 18:44:17 +02:00
Francesco Gatti 664a717353 parse yolo layers 2020-05-30 18:44:02 +02:00
Micaela Verucchi 15e4ab4436 Modify darknetFields_t 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-05-30 18:10:14 +02:00
Francesco Gatti 3951ff2582 layer parser 2020-05-30 18:02:20 +02:00
Micaela Verucchi 26e6ba52a5 Add some fields 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-05-30 18:00:22 +02:00
Francesco Gatti 1755e734b1 parse layer and network 2020-05-30 17:59:00 +02:00
Micaela Verucchi 98173ec15c Add darknetParseFields 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-05-30 17:57:48 +02:00
Micaela Verucchi 8b037781f4 merge 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-05-30 17:33:18 +02:00
Francesco Gatti 179e07e830 parse line by line 2020-05-30 17:27:22 +02:00
Micaela Verucchi cb6dc4f182 Add parseType 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-05-30 17:25:48 +02:00