Commit Graph
100 Commits
Author SHA1 Message Date
2martens 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
2martens 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
2martens 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
2martens 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
2martens 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
2martens df2faaffe9 Updated ssd_keras
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-15 12:30:12 +02:00
2martens 3ec6f9ad0a Updated ssd_keras
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-15 12:18:09 +02:00
2martens 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
2martens 135144efc8 Updated ssd_keras
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-15 11:46:41 +02:00
2martens 779249afff Added save image call after decoding images
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-15 11:31:05 +02:00
2martens e0da7eca04 Save images only from one batch
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-15 11:07:20 +02:00
2martens 5c9eaab6e4 Fixed function calls
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-15 11:03:09 +02:00
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