Commit Graph

737 Commits

Author SHA1 Message Date
Jim Martens facc6c9b1f Updated ssd_keras
Signed-off-by: Jim Martens <github@2martens.de>
2019-08-07 13:45:12 +02:00
Jim Martens 49490aa05b Colour channels were already swapped
Signed-off-by: Jim Martens <github@2martens.de>
2019-08-07 13:45:02 +02:00
Jim Martens b123e90821 Updated ssd_keras
Signed-off-by: Jim Martens <github@2martens.de>
2019-08-07 13:24:49 +02:00
Jim Martens 00b756b10a Added conversion back to RGB before saving images
Signed-off-by: Jim Martens <github@2martens.de>
2019-08-07 13:24:01 +02:00
Jim Martens 05d113a4ba Convert image input to BGR
Signed-off-by: Jim Martens <github@2martens.de>
2019-08-07 13:20:34 +02:00
Jim Martens 29522921f1 Fixed typo
Signed-off-by: Jim Martens <github@2martens.de>
2019-08-05 13:48:21 +02:00
Jim Martens 37d5e32031 Added confidence threshold to config
Signed-off-by: Jim Martens <github@2martens.de>
2019-08-05 13:46:04 +02:00
Jim Martens f4982e364c Made iou threshold actually count
Signed-off-by: Jim Martens <github@2martens.de>
2019-08-05 13:44:34 +02:00
Jim Martens 6576f3d5e1 Removed mapping from 'tap' to fire hydrant
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-25 16:54:06 +02:00
Jim Martens 6034198262 Changed limit for used images to 32
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-25 15:33:41 +02:00
Jim Martens 505ea330fc Added missing creation of visualisation output directory
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-25 15:30:54 +02:00
Jim Martens 45b5fda185 Made visualise work with all tarball ids
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-25 15:21:49 +02:00
Jim Martens 8c31794aad Make sure trajectories are sorted before preparing them for Python
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-25 13:30:01 +02:00
Jim Martens 4a4c23a66e Discard all images without gt again
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-25 13:01:34 +02:00
Jim Martens e213ed62ec Fixed the number of images used for evaluation
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-25 13:01:04 +02:00
Jim Martens 60a9d63a1d Changed code to keep images without GT during evaluation
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-25 12:04:47 +02:00
Jim Martens 84adbe21fe Added code to visualise f1 scores to open set error
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-24 12:52:31 +02:00
Jim Martens 761f6124e0 Prevented overwriting of saved images
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-24 12:04:44 +02:00
Jim Martens 5cba4046a5 Converted category ids to classes for SSD COCO case
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-24 11:59:09 +02:00
Jim Martens 3330ce69e7 Added missing function argument
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-24 11:33:13 +02:00
Jim Martens 856aad8e21 Added code to only take 30 frames of each trajectory in evaluation
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-23 15:39:25 +02:00
Jim Martens 7f14efcb6a Removed obsolete code
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-23 14:21:14 +02:00
Jim Martens f0a996423b Removed debug print
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-23 13:57:13 +02:00
Jim Martens 79c5df1805 Added debug print to visualise_metrics
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-23 13:48:07 +02:00
Jim Martens f950504f20 Added missing network selection
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-23 13:46:15 +02:00
Jim Martens 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
Jim Martens 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
Jim Martens e20167182f Added measurement of usable frames
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-22 16:32:47 +02:00
Jim Martens 7359f52158 Disabled augmentation
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-22 16:09:19 +02:00
Jim Martens 2b656057af Enabled augmentations for training again
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-22 15:20:46 +02:00
Jim Martens 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
Jim Martens 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
Jim Martens 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
Jim Martens 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
Jim Martens 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
Jim Martens 84df7b64fa Really fixed entropy threshold keyword argument
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-18 14:00:17 +02:00
Jim Martens a5555c5f73 Implemented micro and macro averaged metrics
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-18 13:40:19 +02:00
Jim Martens 3b2f9fe4da Fixed entropy threshold keyword argument
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-18 13:27:40 +02:00
Jim Martens 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
Jim Martens 015befd542 Fixed config values
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-18 12:40:06 +02:00
Jim Martens 71928650d3 Converted visualise to clean code standards
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-17 15:41:08 +02:00
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