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
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
28fab9c3e1
Merge remote-tracking branch 'origin/master' into cnet
2021-04-27 14:43:39 +02:00
1de804f98d
Code cleanup and readme fixes
2021-04-09 13:17:41 +05:30
37b2a5bd98
minor modifications
2021-04-08 00:57:58 +05:30
2e92944f1d
Opencv cuda fix
2021-03-31 03:44:51 -07:00
e94e1f7622
tkdnn first patch for windows
2021-03-24 22:26:46 +05:30
6aa8666be5
Commits for msvc 16.9
2021-03-10 12:59:04 +05:30
2d4dececb6
Builds on windows successfully,issues with deserialization and downloading weights
2021-01-26 20:17:43 +04:00
512acd8cba
minor fixes
2021-01-21 09:29:27 +04:00
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
9e1d7b3bb4
Update demo3d with show flag
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-12-09 18:05:54 +01:00
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
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
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
be9e327aef
Fix minor, update readme
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-06-09 21:01:19 +02:00
a0e4e9f209
Modify yolov4_berkeley link for download, clean to merge with master
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-06-01 15:17:53 +02:00
6bdf47bae6
Add 3D demo program
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-05-27 18:05:38 +02:00
377310af50
Add variable batch for detector update
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-05-19 09:32:27 +02:00
23b40de508
Add verbose define
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-05-15 09:41:27 +02:00
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
2e3cb52cff
Add flag to disable visualization and save video
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-05-11 15:08:10 +02:00
533bb48789
Add json detection creation for codalab check
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-05-11 11:57:58 +02:00
85588f7343
warning fix
2020-04-09 20:05:10 +02:00
ed0596a52c
warning fix
2020-04-09 01:00:14 +02:00
ae1d8cd9e6
Refactoring & documentation
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-08 11:20:48 +02:00
326c7e0940
Move extraction of name into function in utils
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-08 10:09:15 +02:00
c36befaf2b
Minor fix on output file name in map demo
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-08 09:35:54 +02:00
c8308963df
Add evaluation to tk::dnn namespace, style fix also
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-06 18:12:38 +02:00
af13e7c954
Add getMemoryUsage function, detection update moved in abstract lass, splitted execution time in pre-inf-post, other minors.
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-01 19:18:43 +02:00
7c55dcb708
Add tp tests for yolo3512 and yolo3tiny512
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-03-26 15:06:49 +01:00
df37e11709
Modify map demo, using abstract class. Move draw function in abstract class
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-03-23 19:32:37 +01:00
bbcc33c0cf
Refactoring for detection NN
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-03-20 21:14:12 +01:00
41b1135fb3
Add support for mobilenet in map_demo
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-03-17 15:08:43 +01:00
1d388ca51a
Add mobilenetv2ssdlite512 test.
...
Works for CuDNN, not for tensorRT.
Modified channels in second convolution for classification headers from 126 to 486, when changing size from 300 to 512.
Added size 512 SSD specs and support to COCO dataset (81 classes, first BACKGROUND due to repo for training).
Refactoring class names.
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-03-17 10:35:02 +01:00
cf7fbadcd2
Add function to download weigths if do no exist, for each test. Add some controls in map demo.
...
Signed-off-by: Francesco Gatti <gattifrancesco@hotmail.it >
2020-03-14 17:02:07 +01:00
db10ebeb34
Fix print on times.csv
...
Signed-off-by: xavier <micaelaverucchi@gmail.com >
2020-03-12 20:18:02 +01:00
4b85a2238a
Add writing results on file for map demo
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-03-10 14:35:25 +01:00
01a42ffe27
Move multiple demo files into one
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-03-03 17:30:35 +01:00
296a6cbc87
Add draw method to Yolo3Detection class.
...
Update yolo demo
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-03-03 17:08:25 +01:00
fe206ea24c
Fix dependencies problems
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-02-27 16:48:10 +01:00
3eb079dd13
Add Mobilenetv2 SSD Lite post and preprocessing, add mobilenet demo
...
Signed-off-by: xavier <micaelaverucchi@gmail.com >
2020-02-26 17:42:20 +01:00
38a1b9dcb2
Add Mobilenet2SSDLite test
...
The new test works both with TensorRT and cuDNN. Preprocessing and
Postprocessing are missing. Add ClippedReLU (for ReLU6), groups for
Conv2d, additional bias for convolution.
Other minors:
-move the timer in the detector to measure all the
processing time for a given frame (both centernet and yolo);
-add int8 flag.
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
Davide Sapienza <sapienza.dav@gmail.com >
2020-02-21 10:45:46 +01:00
7e21b10aee
Add script to download COCO val2017
...
Signed-off-by: xavier <micaelaverucchi@gmail.com >
2020-02-11 14:56:54 +01:00
1936e54870
Change README.md, add config.yaml, refactoring
...
Signed-off-by: xavier <micaelaverucchi@gmail.com >
2020-02-11 11:25:42 +01:00
801b8b5641
Read parameters for mAP from yaml, add yampl-cpp dependency
...
Signed-off-by: xavier <micaelaverucchi@gmail.com >
2020-02-11 10:32:42 +01:00
d601e980f6
Add avg precision, recall, f1score computation, other minors
...
Signed-off-by: xavier <micaelaverucchi@gmail.com >
2020-02-10 18:19:08 +01:00
289a97d06c
Refactoring map_demo, add evaluation.h and evaluation.cpp
...
Signed-off-by: xavier <micaelaverucchi@gmail.com >
2020-02-07 12:47:02 +01:00
f32d8a859b
Add mAP 0.5:0.95, other small fix
...
Signed-off-by: xavier <micaelaverucchi@gmail.com >
2020-02-07 09:09:40 +01:00
d8eeb36d4b
Add mAP computation and demo
...
Signed-off-by: xavier <micaelaverucchi@gmail.com >
2020-02-05 18:39:13 +01:00
Davide Sapienza
perseusdg
Micaela Verucchi
Francesco Gatti