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
8fad424c61
Yolo4-tiny batched
...
fix #59
2020-06-30 19:37:16 +02:00
863e27a04e
Fix minor, update readme
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-06-09 21:01:19 +02:00
2df8bb92c2
merge
2020-06-01 15:34:59 +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
1f4f27d752
Support yolov4
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-28 20:35:30 +02:00
fdfda38fbf
Merge branch 'master' of https://github.com/ceccocats/tkDNN into cnet
2020-04-28 14:43:24 +02:00
a761fef159
Add bdd new tests, add BDD100K_val download, fix yolo3-512 link
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-24 12:39:57 +02:00
1cbf18c21e
test batch
2020-04-23 00:54:03 +02:00
50ada3db60
Remove the optional parameters from the Pooling layer.
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-04-09 19:20:27 +02:00
a8fd4cabf3
Add mnist download, change ERROR_TKDNN into ERROR_TENSORRT
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-09 19:02:25 +02:00
af761786ea
Fix pooling, add return code in each test, add return code handling in test_all_tests script
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-09 15:25:05 +02:00
18aff44c99
Add imuodom download and test in script
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-08 16:50:40 +02:00
01e667d4f9
Improve test_all_tests scripts to run in every mode (FP32, FP16, INT8)
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
Davide Sapienza <sapienza.dav@gmail.com >
2020-04-08 15:29:49 +02:00
56c03e3a12
Minor, uncomment test_all_tests
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-08 15:05:52 +02:00
8c1eb69494
Merge with master works
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
Davide Sapienza <sapienza.dav@gmail.com >
2020-04-08 14:59:42 +02:00
4d1c0f632e
Add scripts folder
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-08 14:24:34 +02:00
Micaela Verucchi
Francesco Gatti
Davide Sapienza