Commit Graph

706 Commits

Author SHA1 Message Date
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 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
Jim Martens 32b23bd55d Overhauled functions to minimize parameter count
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-02 15:31:28 +02:00
Jim Martens 0498bd03c6 Improved formatting
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-02 13:25:22 +02:00
Jim Martens 8aac3ae30f Added doc comments to public functions in cli module
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-02 13:23:10 +02:00
Jim Martens 6e85ab86eb Shortened code for execute action
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-02 13:19:00 +02:00
Jim Martens d58179fa2a Removed obsolete auto-encoder code
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-02 13:10:12 +02:00
Jim Martens afdf7c96aa Clarified initialise config function
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-02 13:00:19 +02:00
Jim Martens ad96c5b156 Removed obsolete local imports
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-02 11:02:56 +02:00
Jim Martens f9c2605603 Simplified evaluate get results function
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-02 10:58:06 +02:00
Jim Martens 686bfcaf40 Fixed evaluation wrt labels and filenames
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-01 13:23:02 +02:00
Jim Martens 9c91f82cc3 Added support for entropy thresholds in evaluate function(s)
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-01 13:05:21 +02:00
Jim Martens b483fad5af Updated ssd_keras
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-01 13:04:45 +02:00
Jim Martens b49087f415 Added yet another empty check
Signed-off-by: Jim Martens <github@2martens.de>
2019-08-29 15:43:08 +02:00