Commit Graph

570 Commits

Author SHA1 Message Date
Micaela Verucchi b4c8c2bbad Add stats
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-07-03 18:45:20 +02:00
Micaela Verucchi 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
Micaela Verucchi 7c2155decf Add weights download link for csresnext50-panet-spp_berkeley ( fix #63 )
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-07-01 11:56:46 +02:00
Micaela Verucchi a5cc4e3eda Add berkeley test, add weights for shelfnets
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-07-01 10:53:24 +02:00
Micaela Verucchi a4dca23111 Minor
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-07-01 10:22:31 +02:00
Francesco Gatti 04602f3952 Yolo4-tiny batched
fix #59
2020-06-30 19:37:16 +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
Micaela Verucchi 3bf9547502 Improve preprocessing
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-06-29 15:11:52 +02:00
Francesco Gatti 6d9beb1ec5 Merge pull request #53 from omaralvarez/master
Fix error when parsing label files with unexpected chars
2020-06-25 10:07:41 +02:00
Micaela Verucchi d25803d438 Post-processing improved
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-06-24 18:46:25 +02:00
Micaela Verucchi 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
Micaela Verucchi 94e558003d Shelfnet works on tensorRT (shortcut need to be fixed)
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-06-23 12:50:24 +02:00
Micaela Verucchi 6fd261f628 Shelfnet works on cuDNN. To test everything else
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-06-22 18:11:19 +02:00
Omar Alvarez 2817ade782 Fix parsing label files with unexpected chars 2020-06-22 13:40:15 +02:00
Micaela Verucchi 9f1e30eaa9 Add shelfnet. Resnet18backbone works
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-06-19 18:55:42 +02:00
Francesco Gatti 1dfc69ba89 viz yolo3 preprocess 2020-06-16 13:19:30 +02:00
Francesco Gatti 07193fc343 dealloc in test.h
fix #36
2020-06-16 12:44:12 +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
Micaela Verucchi 285c77d6dd Imptove relative paths
Signed-off-by:  Micaela Verucchi <micaelaverucchi@gmail.com>
		Francesco Gatti <gattifrancesco@hotmail.it>
2020-06-15 17:15:37 +02:00
Francesco Gatti 6dff675db7 fix wrong commit 2020-06-14 13:03:48 +02:00
Francesco Gatti cbfc8ea4f2 serialize fix 2020-06-14 13:01:45 +02:00
Francesco Gatti 567dc0f75d cmake cudnn fix 2020-06-14 12:57:29 +02:00
Micaela Verucchi ab6d2d1766 Update README
Signed-off-by: Micaela Verucchi <micaela.verucchi@unimore.it>
2020-06-12 11:39:46 +02:00
Micaela Verucchi 8d08f5aade New yolo4_512 download link
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-06-11 20:54:17 +02:00
Micaela Verucchi e094a3e0fc Add script for inference FPS
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-06-11 16:09:01 +02:00
Micaela Verucchi c4e955eab5 Add different yolov4 size tests
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-06-11 12:43:34 +02:00
Micaela Verucchi be9e327aef Fix minor, update readme
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-06-09 21:01:19 +02:00
Francesco Gatti 211eff8ad0 Merge branch 'master' of https://github.com/ceccocats/tkDNN 2020-06-09 20:53:17 +02:00
Francesco Gatti 20303ac32e cudnn8 compile 2020-06-09 20:53:05 +02:00
Micaela Verucchi 3ea23815a4 Update README.md 2020-06-03 11:01:17 +02:00
Francesco Gatti 62e4a3f779 memory release v0.5 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 2f243f26e5 readme update 2020-06-01 19:04:39 +02:00
Francesco Gatti c8ed6d782a all test ok 2020-06-01 16:15:29 +02:00
Francesco Gatti d4e0d07e09 merge 2020-06-01 15:34:59 +02:00
Micaela Verucchi 0548a662f5 Merge branch 'master' of https://github.com/ceccocats/tkDNN into eval 2020-06-01 15:18:29 +02:00
Micaela Verucchi 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
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 d8fbee58d8 Add names files
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-06-01 10:42:06 +02:00
Micaela Verucchi a6eef498fa Check
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-06-01 10:39:13 +02:00
Micaela Verucchi 826fcc97c8 Read classes' names from file
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-06-01 10:34:48 +02:00
Francesco Gatti 6d81473b2a yolo3 parsed ok 2020-05-30 22:11:00 +02:00
Francesco Gatti 0621ed223b Merge branch 'darknetparser' of https://github.com/ceccocats/tkDNN into darknetparser 2020-05-30 19:38:31 +02:00
Francesco Gatti 0ae96a6bc4 darknet parser to be tested on yolo3 2020-05-30 19:38:26 +02:00
Micaela Verucchi d45947eec5 Add group field
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-05-30 19:03:29 +02:00
Micaela Verucchi e5e6654b1d Add fields to darknetParseFields
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-05-30 18:44:17 +02:00
Francesco Gatti 298487d5c5 parse yolo layers 2020-05-30 18:44:02 +02:00