Commit Graph

182 Commits

Author SHA1 Message Date
Micaela Verucchi 83d57436ac 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 4c9b9bf03a yolo4tiny works on tensorRT 🐬 🐬 🐬 🐬
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-06-30 15:52:58 +02:00
Micaela Verucchi 35c0d9fd37 yolov4tiny works on CUDNN
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-06-30 15:19:03 +02:00
Francesco Gatti 6d467e5d93 viz yolo3 preprocess 2020-06-16 13:19:30 +02:00
Francesco Gatti 0f3dbac962 viz yolo3 2020-06-16 12:36:45 +02:00
Micaela Verucchi 1e8ef497d8 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 fec43f35dc fix wrong commit 2020-06-14 13:03:48 +02:00
Francesco Gatti d617062752 serialize fix 2020-06-14 13:01:45 +02:00
Francesco Gatti 235a6838eb all test ok 2020-06-01 16:15:29 +02:00
Francesco Gatti 2df8bb92c2 merge 2020-06-01 15:34:59 +02:00
Micaela Verucchi fd00822a3e 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 af9bb411be darknet parser tested 2020-06-01 14:55:08 +02:00
Francesco Gatti 21518c5eef darknet parse all net to be tested 2020-06-01 12:22:55 +02:00
Micaela Verucchi 702e38a150 Add names files
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-06-01 10:42:06 +02:00
Micaela Verucchi fe07c8fb26 Read classes' names from file
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-06-01 10:34:48 +02:00
Francesco Gatti 4b7e80ae0d yolo3 parsed ok 2020-05-30 22:11:00 +02:00
Francesco Gatti 27a83e79d3 darknet parser to be tested on yolo3 2020-05-30 19:38:26 +02:00
Micaela Verucchi 3e701e7ba3 Add fields to darknetParseFields
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-05-30 18:44:17 +02:00
Francesco Gatti 664a717353 parse yolo layers 2020-05-30 18:44:02 +02:00
Micaela Verucchi 26e6ba52a5 Add some fields
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-05-30 18:00:22 +02:00
Francesco Gatti 1755e734b1 parse layer and network 2020-05-30 17:59:00 +02:00
Micaela Verucchi 98173ec15c Add darknetParseFields
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-05-30 17:57:48 +02:00
Francesco Gatti 179e07e830 parse line by line 2020-05-30 17:27:22 +02:00
Francesco Gatti f9d28c58ca darknet parser interface 2020-05-30 16:58:40 +02:00
Micaela Verucchi ad21cc0b79 Add yolo4_berkeley test
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-05-15 09:48:59 +02:00
Micaela Verucchi b8686186e4 Merge branch 'master' of https://github.com/ceccocats/tkDNN into eval 2020-05-11 11:58:51 +02:00
Micaela Verucchi 5eb0cbfcef 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 54074780cf filter yaw ok 2020-05-03 19:12:10 +02:00
Francesco Gatti b34c013ad7 merge 2020-05-03 15:58:00 +02:00
Francesco Gatti 099aa1accd Imu odom weights 2020-05-03 15:56:30 +02:00
Francesco Gatti eb1d30d1b1 Merge branch 'master' of https://github.com/ceccocats/tkDNN 2020-05-03 15:54:42 +02:00
Francesco Gatti 8d45259b15 SEP model 2020-05-03 15:54:39 +02:00
Francesco Gatti 72dfd92c0a imu odom SEP model 2020-05-03 15:53:35 +02:00
Micaela Verucchi 3be4fe0117 Modify download
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-04-28 20:40:35 +02:00
Micaela Verucchi 1f4f27d752 Support yolov4
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-04-28 20:35:30 +02:00
Francesco Gatti 1d284bf7d6 Merge commit '04bd5d7ff46270c732b76cd9656e22de930e138e' into tree 2020-04-28 14:54:18 +02:00
Micaela Verucchi fdfda38fbf Merge branch 'master' of https://github.com/ceccocats/tkDNN into cnet 2020-04-28 14:43:24 +02:00
Micaela Verucchi dbe54f4b67 Fix shortcutRT plugin, now works with batches, fix yolo3_512 downlaod link
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-04-28 14:34:12 +02:00
Micaela Verucchi a761fef159 Add bdd new tests, add BDD100K_val download, fix yolo3-512 link
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-04-24 12:39:57 +02:00
Francesco Gatti 7597539076 merge tkDNN 2020-04-24 11:18:22 +02:00
Francesco Gatti 2a6f7db4af Flatten batch to be checked 2020-04-23 01:18:27 +02:00
Francesco Gatti 1cbf18c21e test batch 2020-04-23 00:54:03 +02:00
Francesco Gatti 9655c29b94 RouteRt is not used 2020-04-21 19:50:45 +00:00
Francesco Gatti a91b2f46ba batch seems ok in yolo3_berkely
layers to be checked:
DeformableConvRT
FlattenConcatRT
ReshapeRT
RouteRT (dont know why but seems working)
2020-04-21 19:41:40 +00:00
Francesco Gatti ef91439c87 batch size > 1 2020-04-21 19:20:44 +02:00
Micaela Verucchi ec7620ebb6 Move download of weights inside build folder
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-04-17 22:36:25 +02:00
Francesco Gatti 70227b6502 warning fix 2020-04-14 14:26:06 +02:00
Davide Sapienza 50ada3db60 Remove the optional parameters from the Pooling layer.
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
2020-04-09 19:20:27 +02:00
Micaela Verucchi a8fd4cabf3 Add mnist download, change ERROR_TKDNN into ERROR_TENSORRT
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-04-09 19:02:25 +02:00
Davide Sapienza 8a575f6447 Remove the final parameter from the layers
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
2020-04-09 18:48:03 +02:00