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
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
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
Davide Sapienza
1cfa199ee6
Add pre-processing and post-processing stats
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-12-09 19:43:51 +01:00
Davide Sapienza
48ecebe6dd
Add CenterTrack pre, post, visualization and demo.
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-12-07 18:45:29 +01:00
Davide Sapienza
7f65ee0b2a
Merge remote-tracking branch 'origin/master' into cnet
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-12-07 17:45:44 +01:00
Micaela Verucchi
a52e18b6e6
Add shelfnet_mapillary, README_seg, resize of input
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-11-24 12:37:33 +01:00
Micaela Verucchi
a17e7800b9
Add computation of #parameters, #MACC, and max feature map size in the tests
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-11-23 13:07:54 +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
04b4a69107
Merge with master
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-09-14 17:30:02 +02:00
micaela
fa2b3d26cb
Fix typos ( #107 )
...
Signed-off-by: micaela <micaelaverucchi@gmail.com >
2020-09-11 09:13:59 +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
Micaela Verucchi
286e777300
Improved segmentation results, removed resize, code to reorder
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-07-22 19:21:46 +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
Micaela Verucchi
e7779ad773
Add conf_thresh parameter to detectors, add batches handling in map demo
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-07-08 09:52:36 +02:00
tk
a68f45cb4e
Merge with master, add yolov4tiny_512
...
Signed-off-by: tk <micaelaverucchi@gmail.com >
2020-07-06 17:57:50 +02:00
Micaela Verucchi
b4c8c2bbad
Add stats
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-07-03 18:45:20 +02:00
Micaela Verucchi
79cd96de6f
Add resize to original size, writing of segmentation
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-07-02 12:14:20 +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
Micaela Verucchi
3bf9547502
Improve preprocessing
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-06-29 15:11:52 +02:00
Micaela Verucchi
d25803d438
Post-processing improved
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-06-24 18:46:25 +02:00
Micaela Verucchi
082920f3f5
Shelfnet works, also visualization. Postprocessing need to be parallelized
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-06-23 20:01:47 +02:00
Micaela Verucchi
94e558003d
Shelfnet works on tensorRT (shortcut need to be fixed)
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-06-23 12:50:24 +02:00
Micaela Verucchi
6fd261f628
Shelfnet works on cuDNN. To test everything else
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-06-22 18:11:19 +02:00
Francesco Gatti
07193fc343
dealloc in test.h
...
fix #36
2020-06-16 12:44:12 +02:00
Francesco Gatti
1b8f45703f
darknet parser cpp
2020-06-16 12:41:48 +02:00
Francesco Gatti
3d3a2427c9
viz yolo3
2020-06-16 12:36:45 +02:00
Micaela Verucchi
c4e955eab5
Add different yolov4 size tests
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-06-11 12:43:34 +02:00
Francesco Gatti
62e4a3f779
memory release
2020-06-02 12:43:06 +02:00
Francesco Gatti
0458f361b1
release layer wgs and version update
2020-06-01 21:03:37 +02:00
Francesco Gatti
c8ed6d782a
all test ok
2020-06-01 16:15:29 +02:00
Francesco Gatti
d4e0d07e09
merge
2020-06-01 15:34:59 +02:00
Micaela Verucchi
a0e4e9f209
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
18794d52c2
darknet parser tested
2020-06-01 14:55:08 +02:00
Francesco Gatti
d2e2669b6d
darknet parse all net to be tested
2020-06-01 12:22:55 +02:00
Micaela Verucchi
a6eef498fa
Check
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-06-01 10:39:13 +02:00
Micaela Verucchi
826fcc97c8
Read classes' names from file
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-06-01 10:34:48 +02:00
Francesco Gatti
6d81473b2a
yolo3 parsed ok
2020-05-30 22:11:00 +02:00
Francesco Gatti
0621ed223b
Merge branch 'darknetparser' of https://github.com/ceccocats/tkDNN into darknetparser
2020-05-30 19:38:31 +02:00
Francesco Gatti
0ae96a6bc4
darknet parser to be tested on yolo3
2020-05-30 19:38:26 +02:00
Micaela Verucchi
d45947eec5
Add group field
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-05-30 19:03:29 +02:00
Micaela Verucchi
e5e6654b1d
Add fields to darknetParseFields
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-05-30 18:44:17 +02:00