318dfe4b55
merge with master
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2021-07-20 18:38:46 +02:00
69256992a3
Add colormap for each segmentation dataset
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2021-07-20 15:29:26 +02:00
84ff978ccb
Merge with master, all tests passed
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2021-07-20 12:48:41 +02:00
b3c44a86a8
Fix shelfnet berkeley, update readme, minors
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2021-07-20 10:57:53 +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
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
04b4a69107
Merge with master
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-09-14 17:30:02 +02: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
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
594947f301
Update file names map demo
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-07-08 11:27:09 +02:00
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
b4c8c2bbad
Add stats
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-07-03 18:45:20 +02:00
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
a5cc4e3eda
Add berkeley test, add weights for shelfnets
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-07-01 10:53:24 +02:00
d25803d438
Post-processing improved
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-06-24 18:46:25 +02:00
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
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
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
Micaela Verucchi
perseusdg
Francesco Gatti
Davide Sapienza