Commit Graph

196 Commits

Author SHA1 Message Date
Micaela Verucchi 84ff978ccb Merge with master, all tests passed
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2021-07-20 12:48:41 +02:00
Micaela Verucchi b3c44a86a8 Fix shelfnet berkeley, update readme, minors
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2021-07-20 10:57:53 +02:00
Francesco Gatti 6611a91201 merge perseusdg windows10 support 2021-04-24 18:55:26 +02:00
hchandirasekar f12ec3c935 added yolo4x and yolo4-csp from the github repo and download file corrections 2021-03-26 12:18:46 +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
Micaela Verucchi 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
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
Micaela Verucchi b4c8c2bbad Add stats
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-07-03 18:45: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 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 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
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 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 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 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 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 0ae96a6bc4 darknet parser to be tested on yolo3 2020-05-30 19:38:26 +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
Micaela Verucchi 041a5cf65d Add some fields
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-05-30 18:00:22 +02:00
Francesco Gatti 8935d85e91 parse layer and network 2020-05-30 17:59:00 +02:00
Micaela Verucchi 64d22c51f1 Add darknetParseFields
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-05-30 17:57:48 +02:00
Francesco Gatti 548a3dd33c parse line by line 2020-05-30 17:27:22 +02:00
Francesco Gatti 4e1c7a70b1 darknet parser interface 2020-05-30 16:58:40 +02:00
Micaela Verucchi 0a9e01db15 Add yolo4_berkeley test
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-05-15 09:48:59 +02:00
Micaela Verucchi f684a2126e Merge branch 'master' of https://github.com/ceccocats/tkDNN into eval 2020-05-11 11:58:51 +02:00
Micaela Verucchi 533bb48789 Add json detection creation for codalab check
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-05-11 11:57:58 +02:00
Francesco Gatti ca7631250c filter yaw ok 2020-05-03 19:12:10 +02:00
Francesco Gatti 85bbbf42a1 merge 2020-05-03 15:58:00 +02:00
Francesco Gatti df76766890 Imu odom weights 2020-05-03 15:56:30 +02:00
Francesco Gatti 2594e491d4 Merge branch 'master' of https://github.com/ceccocats/tkDNN 2020-05-03 15:54:42 +02:00
Francesco Gatti 5a9ed44b6a SEP model 2020-05-03 15:54:39 +02:00
Francesco Gatti 5ab2e63de4 imu odom SEP model 2020-05-03 15:53:35 +02:00
Micaela Verucchi adb5a693cd Modify download
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-04-28 20:40:35 +02:00
Micaela Verucchi db0f8a4d99 Support yolov4
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-04-28 20:35:30 +02:00
Francesco Gatti e6d435a463 Merge commit '04bd5d7ff46270c732b76cd9656e22de930e138e' into tree 2020-04-28 14:54:18 +02:00