mmr/tkDNN
Archived
1
0
Fork 0
Code Issues 46 Pull Requests 12 Releases 4 Activity
This repository has been archived on 2026-02-22. You can view files and clone it. You cannot open issues or pull requests or push a commit.
Files
f329b89acb277fba5693e2a0415513ee7ae78c70
tkDNN/src
T
History
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
..
kernels
Add mish_yashas
2020-05-30 16:57:53 +02:00
Activation.cpp
Support yolov4
2020-04-28 20:35:30 +02:00
BoundingBox.cpp
Add BoundingBox class
2020-04-08 13:52:29 +02:00
CenternetDetection.cpp
Fix boxes also for Centernet
2020-08-06 15:53:57 +02:00
Conv2d.cpp
Fix patch from @ahmedius2 , tkDNN now supports CUDNN 8.0.1 (Fix #74)
2020-07-16 19:06:54 +02:00
DarknetParser.cpp
Add support for yolov4x-mish.
2020-11-23 11:29:32 +01:00
DeformConv2d.cpp
Fix typos (#107)
2020-09-11 09:13:59 +02:00
Dense.cpp
Fix typos (#107)
2020-09-11 09:13:59 +02:00
evaluation.cpp
Fix typos (#107)
2020-09-11 09:13:59 +02:00
Flatten.cpp
namespace change
2018-12-14 21:55:16 +01:00
Int8BatchStream.cpp
Fix parsing label files with unexpected chars
2020-06-22 13:40:15 +02:00
Int8Calibrator.cpp
Fix the INT8 calibrator sintax
2020-04-08 10:22:03 +02:00
Layer.cpp
memory release
2020-06-02 12:43:06 +02:00
LayerWgs.cpp
Fix typos (#107)
2020-09-11 09:13:59 +02:00
LSTM.cpp
Fix typos (#107)
2020-09-11 09:13:59 +02:00
MobilenetDetection.cpp
Fixed boxes to float, add conf thresh as param
2020-08-05 19:55:10 +02:00
MulAdd.cpp
Fix typos (#107)
2020-09-11 09:13:59 +02:00
Network.cpp
memory release
2020-06-02 12:43:06 +02:00
NetworkRT.cpp
Add support for yolov4x-mish.
2020-11-23 11:29:32 +01:00
NetworkViz.cpp
viz yolo3
2020-06-16 12:36:45 +02:00
Pooling.cpp
Remove the final parameter from the layers
2020-04-09 18:48:03 +02:00
Region.cpp
Fix typos (#107)
2020-09-11 09:13:59 +02:00
Reorg.cpp
yolo3 ok
2018-12-21 15:28:47 +01:00
Reshape.cpp
Remove the final parameter from the layers
2020-04-09 18:48:03 +02:00
Route.cpp
yolov4tiny works on CUDNN
2020-06-30 15:19:03 +02:00
Shortcut.cpp
Fix typos (#107)
2020-09-11 09:13:59 +02:00
Softmax.cpp
Remove the final parameter from the layers
2020-04-09 18:48:03 +02:00
Upsample.cpp
upsample ok
2018-12-20 17:08:31 +01:00
utils.cpp
memory release
2020-06-02 12:43:06 +02:00
Yolo3Detection.cpp
Add support for yolov4x-mish.
2020-11-23 11:29:32 +01:00
Yolo.cpp
Add support for yolov4x-mish.
2020-11-23 11:29:32 +01:00