Commit Graph

417 Commits

Author SHA1 Message Date
Micaela Verucchi 4f3cdd76c8 Add yolo4_mmr test
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-10-23 11:40:55 +02:00
Micaela Verucchi 08f0d3eb34 Update README.md 2020-10-10 13:03:01 +02:00
Micaela Verucchi b10e446a11 Update README with Xavier NX FPS results 15W4Core
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-09-15 14:09:02 +02:00
Micaela Verucchi 6bafa0007f Update README with Xavier NX FPS results
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-09-15 10:50:24 +02:00
micaela a194ba59f6 Fix typos (#107)
Signed-off-by: micaela <micaelaverucchi@gmail.com>
2020-09-11 09:13:59 +02:00
Micaela Verucchi 6d1f6b3881 Merge branch 'master' of https://github.com/ceccocats/tkDNN 2020-08-06 15:54:09 +02:00
Micaela Verucchi c7be331acd Fix boxes also for Centernet
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-08-06 15:53:57 +02:00
Micaela Verucchi 096a67281b Update README.md 2020-08-06 11:05:26 +02:00
Micaela Verucchi 498b163f36 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
Francesco Gatti 8c8341cc61 fix coords convert 2020-07-27 13:52:39 +02:00
Francesco Gatti b4015a8e26 Resolve detection objects pick by prob threshold.
Before this it will only pick the last object with prob > thresh wich is absolutely wrong
Now it picks all the objects with prob > thesh.

fixes #94
2020-07-27 13:45:42 +02:00
Micaela Verucchi e5991378d5 Update README.md 2020-07-17 14:37:10 +02:00
Micaela Verucchi 921ed60cf5 Fix patch from @ahmedius2 , tkDNN now supports CUDNN 8.0.1 (Fix #74)
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
	       Francesco Gatti <gattifrancesco@hotmail.it>
2020-07-16 19:06:54 +02:00
tk 5c7f1c283e Patch for CUDNN 8.0.1
Signed-off-by: tk <micaelaverucchi@gmail.com>
2020-07-16 18:16:09 +02:00
Francesco Gatti 4959f4fa9e Update README.md 2020-07-13 19:51:09 +02:00
Francesco Gatti 79bd3d6f35 docker 2020-07-13 19:50:00 +02:00
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
Francesco Gatti 8fad424c61 Yolo4-tiny batched
fix #59
2020-06-30 19:37:16 +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 562ba06cd8 Merge pull request #53 from omaralvarez/master
Fix error when parsing label files with unexpected chars
2020-06-25 10:07:41 +02:00
Omar Alvarez 98ecd94494 Fix parsing label files with unexpected chars 2020-06-22 13:40:15 +02:00
Francesco Gatti 6d467e5d93 viz yolo3 preprocess 2020-06-16 13:19:30 +02:00
Francesco Gatti 517eaa553b dealloc in test.h
fix #36
2020-06-16 12:44:12 +02:00
Francesco Gatti f9c50b2a49 darknet parser cpp 2020-06-16 12:41:48 +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 ffe332ebb1 cmake cudnn fix 2020-06-14 12:57:29 +02:00
Micaela Verucchi 971de08e9a Update README
Signed-off-by: Micaela Verucchi <micaela.verucchi@unimore.it>
2020-06-12 11:39:46 +02:00
Micaela Verucchi 863e27a04e Fix minor, update readme
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-06-09 21:01:19 +02:00
Francesco Gatti 9d19370bbb Merge branch 'master' of https://github.com/ceccocats/tkDNN 2020-06-09 20:53:17 +02:00
Francesco Gatti b419fdfc04 cudnn8 compile 2020-06-09 20:53:05 +02:00
Micaela Verucchi 95c240223e Update README.md 2020-06-03 11:01:17 +02:00
Francesco Gatti 4f907dde40 memory release 2020-06-02 12:43:06 +02:00
Francesco Gatti 8c79b75227 release layer wgs and version update 2020-06-01 21:03:37 +02:00
Francesco Gatti 8e7a355fab readme update 2020-06-01 19:04:39 +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 7259511ea2 Merge branch 'master' of https://github.com/ceccocats/tkDNN into eval 2020-06-01 15:18:29 +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 9ee8c5eb58 Check
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-06-01 10:39:13 +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 99f126320b Merge branch 'darknetparser' of https://github.com/ceccocats/tkDNN into darknetparser 2020-05-30 19:38:31 +02:00
Francesco Gatti 27a83e79d3 darknet parser to be tested on yolo3 2020-05-30 19:38:26 +02:00