Francesco Gatti
ba8199a030
fix compile error on old tensorrt and remove useless prints
2021-05-16 22:33:45 +02:00
Francesco Gatti
6b8ae1e27c
remove wrong cuda arch that cause big performance gap
...
fix #226
2021-05-16 22:16:21 +02:00
Francesco Gatti
a638592fc7
Merge pull request #199 from rickymedrano/rickymedrano-patch-1
...
Minor Readme Changes
2021-04-24 19:00:58 +02:00
Francesco Gatti
a473a02a44
Merge branch 'perseusdg-master'
2021-04-24 18:55:44 +02:00
Francesco Gatti
6611a91201
merge perseusdg windows10 support
2021-04-24 18:55:26 +02:00
Harshvardhan Chandirasekar
39323ca8d3
Update README.md
2021-04-14 17:22:26 +05:30
Harshvardhan Chandirasekar
f90ee8ab7d
Update CMakeLists.txt
2021-04-09 13:20:44 +05:30
perseusdg
1de804f98d
Code cleanup and readme fixes
2021-04-09 13:17:41 +05:30
perseusdg
37b2a5bd98
minor modifications
2021-04-08 00:57:58 +05:30
perseusdg
cc594f09ef
merge from gitlab
2021-04-08 00:28:00 +05:30
hchandirasekar
2e92944f1d
Opencv cuda fix
2021-03-31 03:44:51 -07:00
Harshvardhan Chandirasekar
5f3ab1472c
Update download_validation.py
2021-03-26 15:09:28 +01:00
hchandirasekar
7018d163ed
python download file
2021-03-26 19:37:44 +05:30
hchandirasekar
f12ec3c935
added yolo4x and yolo4-csp from the github repo and download file corrections
2021-03-26 12:18:46 +05:30
hchandirasekar
f3d1591430
ReadMe.md windows changes
2021-03-25 20:46:12 +05:30
hchandirasekar
44b71ae6f3
ReadMe.md windows changes
2021-03-25 20:44:43 +05:30
hchandirasekar
78859fe191
timer fix
2021-03-25 16:38:53 +05:30
hchandirasekar
e94e1f7622
tkdnn first patch for windows
2021-03-24 22:26:46 +05:30
Harshvardhan Chandirasekar
06787a931f
minor migrations
2021-03-17 23:06:50 +05:30
Harshvardhan Chandirasekar
304ab49897
shared_ptr migrations
2021-03-17 22:26:26 +05:30
hchandirasekar
6aa8666be5
Commits for msvc 16.9
2021-03-10 12:59:04 +05:30
Ricky Medrano
f055341af6
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
4a90314333
minor fixes in test.h
2021-01-26 23:11:58 +05:30
hchandirasekar
2d4dececb6
Builds on windows successfully,issues with deserialization and downloading weights
2021-01-26 20:17:43 +04:00
hchandirasekar
fb52444cdc
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
adac8576b0
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
512acd8cba
minor fixes
2021-01-21 09:29:27 +04:00
perseusdg
56feb54377
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
b8855b9599
Update README
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-11-23 11:34:06 +01:00
Micaela Verucchi
702791e41a
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
86478f9384
Add yolo4_mmr test
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-10-23 11:40:55 +02:00
Micaela Verucchi
a0e7f05a50
Update README.md
2020-10-10 13:03:01 +02:00
Micaela Verucchi
d3372aad31
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
be818d5e3a
Update README with Xavier NX FPS results
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-09-15 10:50:24 +02:00
micaela
fa2b3d26cb
Fix typos ( #107 )
...
Signed-off-by: micaela <micaelaverucchi@gmail.com >
2020-09-11 09:13:59 +02:00
Micaela Verucchi
38106a9495
Merge branch 'master' of https://github.com/ceccocats/tkDNN
2020-08-06 15:54:09 +02:00
Micaela Verucchi
df5443e017
Fix boxes also for Centernet
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-08-06 15:53:57 +02:00
Micaela Verucchi
8bef544bae
Update README.md
2020-08-06 11:05:26 +02:00
Micaela Verucchi
f778e1aa99
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
a5d2d4792a
fix coords convert
2020-07-27 13:52:39 +02:00
Francesco Gatti
3a0802d70c
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
f4970d1e6f
Update README.md
2020-07-17 14:37:10 +02:00
Micaela Verucchi
6a68f19b2c
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
c4aad7fe95
Patch for CUDNN 8.0.1
...
Signed-off-by: tk <micaelaverucchi@gmail.com >
2020-07-16 18:16:09 +02:00
Francesco Gatti
b2df9fc110
Update README.md
2020-07-13 19:51:09 +02:00
Francesco Gatti
b12cf0d7c2
docker
2020-07-13 19:50:00 +02:00
Micaela Verucchi
7c2155decf
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
04602f3952
Yolo4-tiny batched
...
fix #59
2020-06-30 19:37:16 +02:00
Micaela Verucchi
fe2e4eae92
yolo4tiny works on tensorRT 🐬 🐬 🐬 🐬
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-06-30 15:52:58 +02:00
Micaela Verucchi
61aa24c6b7
yolov4tiny works on CUDNN
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-06-30 15:19:03 +02:00