Commit Graph

746 Commits

Author SHA1 Message Date
Jim Martens cd79be4307 Added config option to test only pre-trained version of SSD
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-17 15:25:04 +02:00
Jim Martens f34eb93e71 Added missing functions to evaluate module doc
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-17 13:56:21 +02:00
Jim Martens b2e9a10bdf Removed obsolete import
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-17 13:54:06 +02:00
Jim Martens 1db9fa1c0a Implemented cumulative open set error
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-17 13:53:52 +02:00
Jim Martens 28a0d35d36 Implemented support for predicting with range of entropy thresholds
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-17 12:38:22 +02:00
Jim Martens f270e0add1 Updated ssd_keras
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-17 12:01:07 +02:00
Jim Martens d9edd94b99 Added entropy threshold as configurable option
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-17 11:54:28 +02:00
Jim Martens a8f58a6fc7 Changed tarball id CLI option to string
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-16 13:29:49 +02:00
Jim Martens 0280015a50 Modified measure mapping to select gt path based on tarball id
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-16 13:28:13 +02:00
Jim Martens 13a3cb9170 Save category to names dictionary as well
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-15 17:20:35 +02:00
Jim Martens 83f70b4fdb Fixed mixed up paths
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-15 16:56:51 +02:00
Jim Martens 400b6e4900 Converted measure_mapping to clean code standards
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-15 16:55:23 +02:00
Jim Martens b819013771 Improved formatting
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-15 16:06:24 +02:00
Jim Martens 758c8a7574 Improve filenames in measure_mapping
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-15 16:03:39 +02:00
Jim Martens 5d955d9c3c Reverted previous change
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-15 14:47:59 +02:00
Jim Martens 590b5e95ca Change f1 score to single value
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-15 14:46:09 +02:00
Jim Martens c56440cb49 Added sorting of files after glob to ensure expected order
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-15 14:29:12 +02:00
Jim Martens 4f4ff42d61 Changed code to also save filenames in evaluate
Will need to be changed again if more than one trajectory is evaluated

Signed-off-by: Jim Martens <github@2martens.de>
2019-07-15 14:12:40 +02:00
Jim Martens 83919b1464 Fixed opening of image
Will need to be changed again if more than one trajectory is evaluated

Signed-off-by: Jim Martens <github@2martens.de>
2019-07-15 13:38:19 +02:00
Jim Martens 289275e82f Changed _ssd_evaluate_save_images to save all images
Will need to be changed again if more than one trajectory is evaluated

Signed-off-by: Jim Martens <github@2martens.de>
2019-07-15 13:34:54 +02:00
Jim Martens 2f03be42dd Changed save_ssd_train_images func to work with filenames as well
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-15 13:25:47 +02:00
Jim Martens 5aac93f444 Modified mean average precision to only use classes with ground truth
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-15 12:44:52 +02:00
Jim Martens df2faaffe9 Updated ssd_keras
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-15 12:30:12 +02:00
Jim Martens 3ec6f9ad0a Updated ssd_keras
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-15 12:18:09 +02:00
Jim Martens 13252b37b2 Changed code so that images are only saved on first batch
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-15 11:59:35 +02:00
Jim Martens 135144efc8 Updated ssd_keras
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-15 11:46:41 +02:00
Jim Martens 779249afff Added save image call after decoding images
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-15 11:31:05 +02:00
Jim Martens e0da7eca04 Save images only from one batch
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-15 11:07:20 +02:00
Jim Martens 5c9eaab6e4 Fixed function calls
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-15 11:03:09 +02:00
Jim Martens f66bf31785 Temporarily disabled save images
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-15 10:59:04 +02:00
Jim Martens a521bc4024 Moved print statements to predict loop
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-15 10:57:06 +02:00
Jim Martens 78e00449ec Fixed print statement
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-15 10:54:04 +02:00
Jim Martens 2ecc0023f8 Added debug prints to get shapes
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-15 10:46:15 +02:00
Jim Martens c0d23a731b Eliminate side effects from debug function
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-15 10:43:18 +02:00
Jim Martens 5827d92b8c Added function to save images directly after prediction
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-15 10:34:48 +02:00
Jim Martens d77c27727f Fixed transformation of predictions (2nd)
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-11 17:34:26 +02:00
Jim Martens ba3cfa379e Fixed transformation of predictions
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-11 17:33:08 +02:00
Jim Martens 38229e1848 Only save images from first batch in evaluate
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-11 17:19:56 +02:00
Jim Martens c2be08e0c8 Changed code to save images with predictions not ground truth
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-11 17:13:17 +02:00
Jim Martens 6f6b4b7e34 Fixed debug saving of images to work with predictions
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-11 17:12:54 +02:00
Jim Martens 8ef685dae9 Fixed unbatching of dict (2nd)
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-11 17:09:51 +02:00
Jim Martens 8d1d78bdb6 Fixed unbatching of dict
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-11 17:09:21 +02:00
Jim Martens 1474f30744 Added code to save images during ssd_evaluate
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-11 17:08:00 +02:00
Jim Martens 50476fc3c0 Fixed name of accessed config value
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-11 16:25:18 +02:00
Jim Martens 0d527cf26d Moved general purpose functions up
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-11 16:23:32 +02:00
Jim Martens d53a0cb6b4 Made ssd_evaluate conform to clean code standards
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-11 16:23:01 +02:00
Jim Martens ab9f3f8a94 Removed obsolete constant N_CLASSES
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-11 16:06:05 +02:00
Jim Martens af62d0492c Fixed call of save function
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-11 15:03:39 +02:00
Jim Martens 73e4920cc4 Fixed mode for ssd model
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-11 15:01:49 +02:00
Jim Martens fef7a1eac1 Added missing function argument
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-11 14:59:47 +02:00