Commit Graph

472 Commits

Author SHA1 Message Date
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
Davide Sapienza f78f7bfddc Fix link in readme files
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
2021-07-21 15:35:17 +02:00
Davide Sapienza 25d02ef3ea Update exporting weights README
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
2021-05-11 17:30:24 +02:00
Davide Sapienza fd56e64938 Update the README and split it into several files.
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
2021-05-11 17:02:43 +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 a638592fc7 Merge pull request #199 from rickymedrano/rickymedrano-patch-1
Minor Readme Changes
2021-04-24 19:00:58 +02:00
Francesco Gatti a473a02a44 Merge branch 'perseusdg-master' 2021-04-24 18:55:44 +02:00
Francesco Gatti 6611a91201 merge perseusdg windows10 support 2021-04-24 18:55:26 +02:00
Harshvardhan Chandirasekar 39323ca8d3 Update README.md 2021-04-14 17:22:26 +05:30
Harshvardhan Chandirasekar f90ee8ab7d Update CMakeLists.txt 2021-04-09 13:20:44 +05:30
perseusdg 1de804f98d Code cleanup and readme fixes 2021-04-09 13:17:41 +05:30
perseusdg 37b2a5bd98 minor modifications 2021-04-08 00:57:58 +05:30
perseusdg cc594f09ef merge from gitlab 2021-04-08 00:28:00 +05:30
hchandirasekar 2e92944f1d Opencv cuda fix 2021-03-31 03:44:51 -07:00
Harshvardhan Chandirasekar 5f3ab1472c Update download_validation.py 2021-03-26 15:09:28 +01:00
hchandirasekar 7018d163ed python download file 2021-03-26 19:37:44 +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 f3d1591430 ReadMe.md windows changes 2021-03-25 20:46:12 +05:30
hchandirasekar 44b71ae6f3 ReadMe.md windows changes 2021-03-25 20:44:43 +05:30
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
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 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