Commit Graph

447 Commits

Author SHA1 Message Date
Francesco Gatti 50cb2ca253 fix compile error on old tensorrt and remove useless prints 2021-05-16 22:33:45 +02:00
Francesco Gatti c2fec5cc8f remove wrong cuda arch that cause big performance gap
fix #226
2021-05-16 22:16:21 +02:00
Francesco Gatti 471d4b27d0 Merge pull request #199 from rickymedrano/rickymedrano-patch-1
Minor Readme Changes
2021-04-24 19:00:58 +02:00
Francesco Gatti 3d0742ad54 Merge branch 'perseusdg-master' 2021-04-24 18:55:44 +02:00
Francesco Gatti 1a5d68ae39 merge perseusdg windows10 support 2021-04-24 18:55:26 +02:00
Harshvardhan Chandirasekar b006e16649 Update README.md 2021-04-14 17:22:26 +05:30
Harshvardhan Chandirasekar 01fa9b668b Update CMakeLists.txt 2021-04-09 13:20:44 +05:30
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
Harshvardhan Chandirasekar 516017cc8f Update download_validation.py 2021-03-26 15:09:28 +01:00
hchandirasekar 06b7538da1 python download file 2021-03-26 19:37:44 +05:30
hchandirasekar a2b0dd62d1 added yolo4x and yolo4-csp from the github repo and download file corrections 2021-03-26 12:18:46 +05:30
hchandirasekar 557fb732a5 ReadMe.md windows changes 2021-03-25 20:46:12 +05:30
hchandirasekar 3b2df9ee85 ReadMe.md windows changes 2021-03-25 20:44:43 +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 0180f7830b Commits for msvc 16.9 2021-03-10 12:59:04 +05:30
Ricky Medrano 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
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
hchandirasekar 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
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 4fa008bd85 Update README
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-11-23 11:34:06 +01: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 Verucchi 4f3cdd76c8 Add yolo4_mmr test
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-10-23 11:40:55 +02:00
Micaela Verucchi 08f0d3eb34 Update README.md 2020-10-10 13:03:01 +02:00
Micaela Verucchi 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
Micaela Verucchi 6bafa0007f Update README with Xavier NX FPS results
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-09-15 10:50:24 +02:00
micaela a194ba59f6 Fix typos (#107)
Signed-off-by: micaela <micaelaverucchi@gmail.com>
2020-09-11 09:13:59 +02:00
Micaela Verucchi 6d1f6b3881 Merge branch 'master' of https://github.com/ceccocats/tkDNN 2020-08-06 15:54:09 +02:00
Micaela Verucchi c7be331acd Fix boxes also for Centernet
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-08-06 15:53:57 +02:00
Micaela Verucchi 096a67281b Update README.md 2020-08-06 11:05:26 +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
Francesco Gatti 8c8341cc61 fix coords convert 2020-07-27 13:52:39 +02:00
Francesco Gatti 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
Micaela Verucchi e5991378d5 Update README.md 2020-07-17 14:37:10 +02:00
Micaela Verucchi 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
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 4959f4fa9e Update README.md 2020-07-13 19:51:09 +02:00
Francesco Gatti 79bd3d6f35 docker 2020-07-13 19:50:00 +02:00
Micaela Verucchi 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
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