mmr/tkDNN
Archived
1
0
Fork 0
Code Issues 46 Pull Requests 12 Releases 4 Activity
530 Commits 11 Branches 5 Tags
31af5396bb96db5867b2e12b94c6b7b1cbd4a292
Commit Graph

269 Commits

Author SHA1 Message Date
Adam Cuellar 31af5396bb Merge branch 'ceccocats:master' into master 2021-11-09 16:50:40 -05:00
Micaela Verucchi d6fb6c6af4 Fix max elem (remove thrust) for segmentation 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2021-11-04 19:28:55 +01:00
Cuellar, Adam T 000af24d29 Add swish 2021-07-27 13:09:43 +00:00
Davide Sapienza 0292e21c33 Merge remote-tracking branch 'origin/master' into cnet 2021-07-22 17:03:23 +02:00
Davide Sapienza b9f82510d3 Add demo_utils. 
This commit splits the utils file into two different files:
utils and demo_utils, to overcome some opencv problems caused
by demo_utils includes.

Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2021-07-22 10:45:18 +02:00
Davide Sapienza 1216e8bb74 Add camera calibration file reading for demo3d and demoTracker. 
Some minor fixes.

Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2021-07-21 16:14:53 +02:00
Davide Sapienza be2d361ac6 Update 3d detection 
This commit updates the CenternetDetection3d class and fixes
some bugs.

It also removes the resnet101 network for CenterNet 3D.
This network doesn't exist.

Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2021-07-21 15:51:13 +02:00
Micaela Verucchi 318dfe4b55 merge with master 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2021-07-20 18:38:46 +02:00
Micaela Verucchi 69256992a3 Add colormap for each segmentation dataset 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2021-07-20 15:29:26 +02:00
Micaela Verucchi 84ff978ccb Merge with master, all tests passed 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2021-07-20 12:48:41 +02:00
Micaela Verucchi b3c44a86a8 Fix shelfnet berkeley, update readme, minors 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2021-07-20 10:57:53 +02:00
Francesco Gatti ba8199a030 fix compile error on old tensorrt and remove useless prints 2021-05-16 22:33:45 +02:00
Davide Sapienza 34c1c3d577 Update cnet branch. 
This commit splits the demo3D in two demo: one for the 3D object
detection and one for the tracking.

It renames the files related to CenterTrack.

It adds a new parameter to select the tracker mode (2D or 3D).

Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2021-05-11 16:17:23 +02:00
Davide Sapienza 0dc96d2a9e Improve CenterTrack. 
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2021-05-04 17:59:20 +02:00
Fabio Bagni f8327e2dac Merge with cnet 2021-05-04 11:50:08 +02:00
Fabio Bagni 10f39d1055 Fix tracker for batch size > 1 2021-05-04 11:21:05 +02:00
Davide Sapienza d1ae1791d9 Fix a bug in the draw function of CenterTrack. 
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2021-05-03 18:56:07 +02:00
Davide Sapienza ff6e0e010a Fix a bug with batch > 1 
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2021-04-30 22:26:33 +02:00
Davide Sapienza 2367519799 Add the calibration matrix reading for CenterTrack 
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2021-04-30 17:10:51 +02:00
Davide Sapienza be6ad27c11 Batch size > 1 for the 3D demo. 
This commit lets to use differtent batch size for 3D CenterNet
and CenterTrack.

Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2021-04-29 11:13:24 +02:00
Davide Sapienza 28fab9c3e1 Merge remote-tracking branch 'origin/master' into cnet 2021-04-27 14:43:39 +02:00
Francesco Gatti 6611a91201 merge perseusdg windows10 support 2021-04-24 18:55:26 +02:00
perseusdg 1de804f98d Code cleanup and readme fixes 2021-04-09 13:17:41 +05:30
perseusdg cc594f09ef merge from gitlab 2021-04-08 00:28:00 +05:30
hchandirasekar f12ec3c935 added yolo4x and yolo4-csp from the github repo and download file corrections 2021-03-26 12:18:46 +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
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 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 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 df5443e017 Fix boxes also for Centernet 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-08-06 15:53:57 +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 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
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