Commit Graph

737 Commits

Author SHA1 Message Date
Jim Martens 0eb1fee649 Completed documentation
Signed-off-by: Jim Martens <github@2martens.de>
2019-10-01 17:19:32 +02:00
Jim Martens abd702411d Added coding info to scenenet protobuf file
Signed-off-by: Jim Martens <github@2martens.de>
2019-10-01 17:06:11 +02:00
Jim Martens ae5170bb72 Updated README for PyPi release
Signed-off-by: Jim Martens <github@2martens.de>
2019-10-01 17:03:55 +02:00
Jim Martens 8f78426a39 Made visualise metrics compatible with older result files
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-26 12:26:11 +02:00
Jim Martens dbe7b5d844 Added per-class scores to visualise metrics
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-26 12:13:05 +02:00
Jim Martens 6b7c3c9172 Renamed debug module to more fitting 'plotting'
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-24 12:26:43 +02:00
Jim Martens 48ee5466fb Fixed preparation of predictions for evaluate
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-16 13:17:31 +02:00
Jim Martens d24c622d86 Moved GT colour to dark blue for better readability
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-16 12:46:44 +02:00
Jim Martens 220e0927e0 Moved GT colour to green
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-16 12:24:58 +02:00
Jim Martens d16edef94b Added ability for suffix
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-16 12:22:12 +02:00
Jim Martens 06be98f59f Prevented cut off label text
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-16 12:15:39 +02:00
Jim Martens 34d7128a1c Made labels more visible
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-16 12:10:34 +02:00
Jim Martens 5f7870b92f Fixed check for nesting level of colors array
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-16 12:10:16 +02:00
Jim Martens 4fc8bb6f8c Added function to draw single bbox figure
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-16 11:39:31 +02:00
Jim Martens 554ccc84f9 Moved code into separate functions
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-16 11:04:43 +02:00
Jim Martens 8230488031 Save predictions directly before observations are grouped
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-14 13:43:28 +02:00
Jim Martens 6a990b204a Fixed per-class grouping of predictions
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-13 13:48:58 +02:00
Jim Martens 75f4e4cb1e Optimised position of saving predictions after observations are grouped
Before the entropy threshold is applied, there is no difference
between the entropy thresholds. Therefore, the predictions should only
be saved once.

Signed-off-by: Jim Martens <github@2martens.de>
2019-09-13 13:05:09 +02:00
Jim Martens d877da3ef3 Improved readability of code
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-13 13:00:14 +02:00
Jim Martens 01f43651b5 Fixed ssd train function
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-13 12:35:52 +02:00
Jim Martens 5bde170138 Updated ssd_keras
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-13 12:25:24 +02:00
Jim Martens dd2d0f407b Added config option to enable/disable dropout
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-13 12:25:13 +02:00
Jim Martens 53022bf1a9 Added saving of predictions before entropy threshold
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-13 12:15:54 +02:00
Jim Martens 681d23f345 Removed obsolete code
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-13 11:56:26 +02:00
Jim Martens 64300c3842 Reduced parameter size for predict function
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-13 11:55:19 +02:00
Jim Martens 90d64ae7c7 Added use_nms config option
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-13 11:47:18 +02:00
Jim Martens 3d78309603 Made NMS configurable
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-13 11:41:39 +02:00
Jim Martens 343d832545 Added no NMS run to combined graphs
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-12 17:53:15 +02:00
Jim Martens 5e50be22dc Updated predictions glob string for new suffixes
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-12 17:41:00 +02:00
Jim Martens b861270e51 Save predictions directly after entropy threshold is applied as well
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-12 15:24:33 +02:00
Jim Martens c0e7b434dc Added suffix to save function
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-12 15:23:45 +02:00
Jim Martens aa114c7237 Added _env to gitignore
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-12 12:06:02 +02:00
Jim Martens c3286b7169 Set limits to figures
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-11 12:07:51 +02:00
Jim Martens 56a635e5e6 Dealt with different keys in stored metrics files
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-11 11:35:18 +02:00
Jim Martens 9d82b9bd5c Removed obsolete import
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-11 11:34:55 +02:00
Jim Martens 60b40bc53e Removed unnecessary argument
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-11 11:29:36 +02:00
Jim Martens 2c226e9975 Added CLI command to visualise a combined graph
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-11 11:25:47 +02:00
Jim Martens 6f543b947e Removed entropy from structured array
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-09 09:32:48 +02:00
Jim Martens a91ff8cc22 Fixed filtering for Bayesian SSD
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-09 08:56:08 +02:00
Jim Martens 909d8f7bef Fixed Bayesian NMS
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-08 22:57:10 +02:00
Jim Martens ec03b0df4e Added NMS for Bayesian SSD
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-08 22:00:34 +02:00
Jim Martens a41f607ca9 Changed filtering to occur per-class
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-08 21:47:00 +02:00
Jim Martens aaaea5f942 Added if clause to check for empty instance
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-08 21:45:32 +02:00
Jim Martens eef8839657 Unified naming of evaluation results
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-04 12:12:30 +02:00
Jim Martens 0ae13a2fba Added support for entropy thresholds in visualise metrics
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-04 11:55:19 +02:00
Jim Martens de8001c2d3 Save json with indent level 2
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-04 11:41:54 +02:00
Jim Martens 6b56ba9aba Changed annotation meta file to speed up saving images
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-04 11:39:15 +02:00
Jim Martens 5df06bbfa6 Fixed serialisation of scores
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-03 12:44:35 +02:00
Jim Martens 66d81f3362 Fixed path names
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-03 12:44:21 +02:00
Jim Martens e1ff56c5d3 Added missing metrics
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-03 12:16:31 +02:00