Commit Graph

715 Commits

Author SHA1 Message Date
f0a996423b Removed debug print
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-23 13:57:13 +02:00
79c5df1805 Added debug print to visualise_metrics
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-23 13:48:07 +02:00
f950504f20 Added missing network selection
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-23 13:46:15 +02:00
53472a8342 Added code to visualise metrics
For starters, only precision/recall is visualised

Signed-off-by: Jim Martens <github@2martens.de>
2019-07-23 13:14:22 +02:00
d6047665ef Added functionality to use SSD with COCO data set
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-23 11:21:06 +02:00
e20167182f Added measurement of usable frames
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-22 16:32:47 +02:00
7359f52158 Disabled augmentation
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-22 16:09:19 +02:00
2b656057af Enabled augmentations for training again
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-22 15:20:46 +02:00
d1dadabc0d Removed guard clause and use 0 as precision/recall if no class predictions
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-18 15:06:34 +02:00
1d21da8ca1 Fixed dtype of micro and macro precision/recall numpy arrays
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-18 15:02:26 +02:00
8897a65b3d Added additional guard clause for cases with no true/false positive entries
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-18 14:58:06 +02:00
60fb13cddf Added debug print for shape of true positives
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-18 14:53:44 +02:00
797323f6e9 Added debug print for diff to largest class
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-18 14:49:35 +02:00
84df7b64fa Really fixed entropy threshold keyword argument
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-18 14:00:17 +02:00
a5555c5f73 Implemented micro and macro averaged metrics
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-18 13:40:19 +02:00
3b2f9fe4da Fixed entropy threshold keyword argument
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-18 13:27:40 +02:00
1a0398191a Removed checkpoint path from prepared paths for ssd test
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-18 12:46:57 +02:00
015befd542 Fixed config values
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-18 12:40:06 +02:00
71928650d3 Converted visualise to clean code standards
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-17 15:41:08 +02:00
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
f34eb93e71 Added missing functions to evaluate module doc
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-17 13:56:21 +02:00
b2e9a10bdf Removed obsolete import
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-17 13:54:06 +02:00
1db9fa1c0a Implemented cumulative open set error
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-17 13:53:52 +02:00
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
f270e0add1 Updated ssd_keras
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-17 12:01:07 +02:00
d9edd94b99 Added entropy threshold as configurable option
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-17 11:54:28 +02:00
a8f58a6fc7 Changed tarball id CLI option to string
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-16 13:29:49 +02:00
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
13a3cb9170 Save category to names dictionary as well
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-15 17:20:35 +02:00
83f70b4fdb Fixed mixed up paths
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-15 16:56:51 +02:00
400b6e4900 Converted measure_mapping to clean code standards
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-15 16:55:23 +02:00
b819013771 Improved formatting
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-15 16:06:24 +02:00
758c8a7574 Improve filenames in measure_mapping
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-15 16:03:39 +02:00
5d955d9c3c Reverted previous change
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-15 14:47:59 +02:00
590b5e95ca Change f1 score to single value
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-15 14:46:09 +02:00
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
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
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
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
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
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
df2faaffe9 Updated ssd_keras
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-15 12:30:12 +02:00
3ec6f9ad0a Updated ssd_keras
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-15 12:18:09 +02:00
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
135144efc8 Updated ssd_keras
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-15 11:46:41 +02:00
779249afff Added save image call after decoding images
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-15 11:31:05 +02:00
e0da7eca04 Save images only from one batch
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-15 11:07:20 +02:00
5c9eaab6e4 Fixed function calls
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-15 11:03:09 +02:00
f66bf31785 Temporarily disabled save images
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-15 10:59:04 +02:00
a521bc4024 Moved print statements to predict loop
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-15 10:57:06 +02:00