mmr/tkDNN
Archived
1
0
Fork 0
Code Issues 46 Pull Requests 12 Releases 4 Activity
608 Commits 11 Branches 5 Tags
matthewgiarra/dockerfile.base-add-libyaml-cpp-dev
Commit Graph

608 Commits

This Branch
This Branch
All Branches
Author SHA1 Message Date
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
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
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
Micaela Verucchi a5cc4e3eda Add berkeley test, add weights for shelfnets 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-07-01 10:53:24 +02:00
Micaela Verucchi a4dca23111 Minor 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-07-01 10:22:31 +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
Francesco Gatti 6d9beb1ec5 Merge pull request #53 from omaralvarez/master 
Fix error when parsing label files with unexpected chars
 2020-06-25 10:07:41 +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