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
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
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
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
Omar Alvarez
2817ade782
Fix parsing label files with unexpected chars
2020-06-22 13:40:15 +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
Francesco Gatti
cbfc8ea4f2
serialize fix
2020-06-14 13:01:45 +02:00
Francesco Gatti
20303ac32e
cudnn8 compile
2020-06-09 20:53:05 +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
d4e0d07e09
merge
2020-06-01 15:34:59 +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
d936e5f740
Add mish_yashas
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-05-30 16:57:53 +02:00
Davide Sapienza
ba8c282384
Add 3D CenterNet detection class
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-05-27 18:03:09 +02:00
Micaela Verucchi
854e1d909a
Fix original size
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-05-19 16:33:45 +02:00
Micaela Verucchi
5d01a3f629
Fix Centernet postprocessing
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-05-14 17:13:38 +02:00
Micaela Verucchi
40456592fc
Adapt detection classes to use batches, adapt demos, update README
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-05-14 16:41:51 +02:00