Commit Graph
100 Commits
Author SHA1 Message Date
2martens f66bf31785 Temporarily disabled save images
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-15 10:59:04 +02:00
2martens a521bc4024 Moved print statements to predict loop
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-15 10:57:06 +02:00
2martens 78e00449ec Fixed print statement
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-15 10:54:04 +02:00
2martens 2ecc0023f8 Added debug prints to get shapes
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-15 10:46:15 +02:00
2martens c0d23a731b Eliminate side effects from debug function
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-15 10:43:18 +02:00
2martens 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
2martens d77c27727f Fixed transformation of predictions (2nd)
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-11 17:34:26 +02:00
2martens ba3cfa379e Fixed transformation of predictions
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-11 17:33:08 +02:00
2martens 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
2martens 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
2martens 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
2martens 8ef685dae9 Fixed unbatching of dict (2nd)
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-11 17:09:51 +02:00
2martens 8d1d78bdb6 Fixed unbatching of dict
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-11 17:09:21 +02:00
2martens 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
2martens 50476fc3c0 Fixed name of accessed config value
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-11 16:25:18 +02:00
2martens 0d527cf26d Moved general purpose functions up
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-11 16:23:32 +02:00
2martens 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
2martens ab9f3f8a94 Removed obsolete constant N_CLASSES
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-11 16:06:05 +02:00
2martens af62d0492c Fixed call of save function
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-11 15:03:39 +02:00
2martens 73e4920cc4 Fixed mode for ssd model
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-11 15:01:49 +02:00
2martens fef7a1eac1 Added missing function argument
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-11 14:59:47 +02:00
2martens e8a82fbfaf Updated ssd_train function to use new signature of predict function
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-11 14:56:42 +02:00
2martens 6c431e5e70 Moved public train function before the private functions
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-11 14:55:20 +02:00
2martens fbcf2c261c Changed predict function to conform to clean code standards
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-11 14:52:59 +02:00
2martens 718467e43c Moved private function to bottom
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-11 12:32:37 +02:00
2martens bc8c8b1c6f Removed not needed return doc
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-11 12:31:03 +02:00
2martens aa6960ceb2 Fixed type hint
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-11 12:30:19 +02:00
2martens 2d19ccf4a5 Changed ssd_train to use new compile_model function
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-11 12:30:08 +02:00
2martens 06548dc8e5 Fixed typo
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-11 12:29:31 +02:00
2martens b30b332c7c Made ssd_test conform to clean code principles
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-11 12:29:06 +02:00
2martens ba2bd776ea Improved formatting
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-11 12:22:10 +02:00
2martens 0ae33130ad Made image size parameter more compatible with clean code
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-11 12:22:01 +02:00
2martens 9fa525ebe3 Added extra functions to compile model and get loss function
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-11 12:20:34 +02:00
2martens a848c12d64 Cleaned up ssd module and removed no longer needed code
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-11 11:24:06 +02:00
2martens 253a507c90 Fixed indexing of instance array
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-10 17:36:33 +02:00
2martens b5058d9ee9 Updated debug print
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-10 17:33:15 +02:00
2martens 82689ba6ec Added debug print
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-10 17:31:02 +02:00
2martens ce87be6019 Added conversion to numpy array
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-10 17:29:35 +02:00
2martens 25de799d80 Changed internal coordinates to corners
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-10 17:28:46 +02:00
2martens a685bba4fa Properly reversed encoding of labels to visualize them
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-10 17:26:16 +02:00
2martens a97b9d8d21 Set internal coordinates to "corners"
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-10 17:12:15 +02:00
2martens 4c4c05d383 Made it possible to save images without preventing training
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-10 16:29:39 +02:00
2martens d517c343be Modified code to save images before and after encoding
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-10 16:16:08 +02:00
2martens c2d8da39c8 Fixed calculation of bbox coordinates
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-10 16:11:14 +02:00
2martens 4f6611ec6c Updated train_keras to work with SSD model directly
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-10 16:05:31 +02:00
2martens 67c098b0d6 Added conversion to int for class_id
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-10 16:01:18 +02:00
2martens aaec64786c Added debug print of class_id
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-10 15:55:01 +02:00
2martens 7969716e6d Made function compatible with encoded and decoded labels
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-10 15:51:20 +02:00
2martens e6370ee92e Fixed function call
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-10 15:46:08 +02:00
2martens 093cbe472e Changed code to save debug images from processed but not encoded labels
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-10 15:38:06 +02:00
2martens 21cef4a09e Made ssd_train function compatible with clean code standards
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-10 15:25:18 +02:00
2martens ed3fec54af Added function to retrieve SSD model
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-10 15:24:49 +02:00
2martens b01cda512a Added some config values
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-10 15:23:30 +02:00
2martens d8d7b51655 Made load_scenenet_data more compatible with clean code principles
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-10 14:52:55 +02:00
2martens 91b7febabb Changed train function to be an integration
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-10 12:15:17 +02:00
2martens a90c6c7c92 Modified debug module to conform to clean code standards
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-10 12:12:13 +02:00
2martens 4cf6723c28 Added ability to use custom string for debug save of train images
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-10 12:02:10 +02:00
2martens b926d11b84 Fixed path for train images
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-10 11:59:34 +02:00
2martens db53c35929 Skip background boxes when saving training images
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-09 15:34:05 +02:00
2martens 90fe1be4a6 Updated ssd_keras
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-09 14:21:57 +02:00
2martens a0a976f9d7 Fixed saving debug images
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-09 14:21:38 +02:00
2martens bdd0862d68 Added guardian function to catch case with zero ground truth
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-09 13:45:01 +02:00
2martens c6e7dd3c50 Accounted for entropy in evaluation code
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-09 13:41:11 +02:00
2martens 4cf782f7ab Fixed formatting
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-09 13:35:47 +02:00
2martens b73cffb8c4 Removed empty line
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-09 13:35:18 +02:00
2martens 7b3367739a Added evaluation path option to config
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-09 13:33:10 +02:00
2martens 9069abcd57 Updated ssd_keras
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-09 13:30:04 +02:00
2martens e25dfe86ca Updated ssd_keras
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-09 13:27:12 +02:00
2martens 3418cb3861 Removed debug print
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-09 13:15:06 +02:00
2martens 3fd7ba7373 Added conversion to numpy array
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-09 13:14:24 +02:00
2martens c96783b384 Updated debug print
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-09 13:13:01 +02:00
2martens 412f0ac453 Added debug print for predictions
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-09 13:00:23 +02:00
2martens 38ce65d367 Fixed not-compiled model
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-09 12:53:15 +02:00
2martens 9b13c5cf52 Fixed function call
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-09 12:25:11 +02:00
2martens 01a09ba474 Fixed SSD prediction
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-09 12:23:27 +02:00
2martens 311be36acf Fixed requesting non-existent property
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-09 12:21:08 +02:00
2martens b123d676bc Predictor sizes only required for training
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-09 12:20:37 +02:00
2martens 4bdd069650 Added custom objects to recover saved model
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-09 12:13:41 +02:00
2martens f5c8e9483a Removed obsolete CLI options
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-09 12:05:23 +02:00
2martens 1d372aba59 Use one trajectory only for prediction
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-09 12:03:38 +02:00
2martens 0c7e96ee9d Disabled early stopping
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-09 11:27:45 +02:00
2martens 320ba40389 Fixed paths
- some still used no longer existing path arguments
- in most cases the directories were not created when needed

Signed-off-by: Jim Martens <github@2martens.de>
2019-07-09 11:22:11 +02:00
2martens 94a4c69896 Changed paths to use config provided paths
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-07 13:55:01 +02:00
2martens 3e9c90f201 Added option "ssd_forward_passes_per_image"
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-07 13:50:59 +02:00
2martens 4886fda32e Added missing argument for function call
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-07 13:43:19 +02:00
2martens ecbd7ea00b Fixed weights path
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-07 13:42:12 +02:00
2martens 322f2c51ab Fixed save dir for models
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-07 13:39:45 +02:00
2martens e3a05eb5fc Truly implemented messaging for main module
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-04 19:39:43 +02:00
2martens 1aa19a27ff Modified config part in CLI module to use messaging
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-04 19:27:17 +02:00
2martens 83083479f7 Modified main file to conform to messaging principle
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-04 18:49:44 +02:00
2martens 89adca3ac4 Removed obsolete import
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-04 17:11:29 +02:00
2martens 77e86f85e4 Added additional hyper parameters to config
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-04 17:11:13 +02:00
2martens da2b788447 Only save summaries if configured to do so
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-04 17:08:06 +02:00
2martens 662c5ec7af Extracted debug functionality into separate module
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-04 17:04:14 +02:00
2martens eda3496f04 Updated the SSD training with new config
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-04 16:43:44 +02:00
2martens 01f5de6d6c Changed config get function to return correct type
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-04 16:33:59 +02:00
2martens a2fe3e860a Added parameters for batch size and ssd image size
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-04 16:30:29 +02:00
2martens b6beebd0c1 Updated CLI arguments for prepare action
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-04 16:27:43 +02:00
2martens 3354a033fe Updated prepare action to use new config
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-04 16:25:57 +02:00
2martens f6c66d1d34 Reordered functions in cli module
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-04 16:15:50 +02:00