Commit Graph

570 Commits

Author SHA1 Message Date
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
Micaela Verucchi 4b3731928c Add yolo4_320_coco2 (pedestrian and stop sign)
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2021-02-25 09:57:10 +01:00
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
Davide Sapienza 59b0f434a7 Fix a bug in the 3D bounding boxes.
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
2020-12-14 15:32:35 +01: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 9e1d7b3bb4 Update demo3d with show flag
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
2020-12-09 18:05:54 +01:00
Davide Sapienza dbc052865c Fix a wrong path in CenternetDetection3DTrack.cpp
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
2020-12-09 18:04:42 +01:00
Davide Sapienza 4543df8533 Update readme.
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
2020-12-07 18:56:07 +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 9f10c3f6e2 Add CenterTrack based on dla34.
The implemented network refers to the nuScenes_3Dtracking.pth model.

Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
2020-12-07 17:59:27 +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 8fb5772ad9 Update README_seg.md 2020-11-24 13:03:02 +01:00
Micaela Verucchi 43d213ba36 Update README_seg.md 2020-11-24 12:44:36 +01:00
Micaela Verucchi f137bcb694 Update README_seg
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-11-24 12:39:37 +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 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 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 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 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
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 594947f301 Update file names map demo
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-07-08 11:27: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 65e2074dda Merge branch 'eval' of https://github.com/ceccocats/tkDNN into eval 2020-07-06 17:58:08 +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