Commit Graph

746 Commits

Author SHA1 Message Date
Jim Martens 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
Jim Martens 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
Jim Martens 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
Jim Martens 718467e43c Moved private function to bottom
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-11 12:32:37 +02:00
Jim Martens bc8c8b1c6f Removed not needed return doc
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-11 12:31:03 +02:00
Jim Martens aa6960ceb2 Fixed type hint
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-11 12:30:19 +02:00
Jim Martens 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
Jim Martens 06548dc8e5 Fixed typo
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-11 12:29:31 +02:00
Jim Martens 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
Jim Martens ba2bd776ea Improved formatting
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-11 12:22:10 +02:00
Jim Martens 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
Jim Martens 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
Jim Martens 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
Jim Martens 253a507c90 Fixed indexing of instance array
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-10 17:36:33 +02:00
Jim Martens b5058d9ee9 Updated debug print
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-10 17:33:15 +02:00
Jim Martens 82689ba6ec Added debug print
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-10 17:31:02 +02:00
Jim Martens ce87be6019 Added conversion to numpy array
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-10 17:29:35 +02:00
Jim Martens 25de799d80 Changed internal coordinates to corners
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-10 17:28:46 +02:00
Jim Martens 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
Jim Martens a97b9d8d21 Set internal coordinates to "corners"
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-10 17:12:15 +02:00
Jim Martens 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
Jim Martens 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
Jim Martens c2d8da39c8 Fixed calculation of bbox coordinates
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-10 16:11:14 +02:00
Jim Martens 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
Jim Martens 67c098b0d6 Added conversion to int for class_id
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-10 16:01:18 +02:00
Jim Martens aaec64786c Added debug print of class_id
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-10 15:55:01 +02:00
Jim Martens 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
Jim Martens e6370ee92e Fixed function call
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-10 15:46:08 +02:00
Jim Martens 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
Jim Martens 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
Jim Martens ed3fec54af Added function to retrieve SSD model
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-10 15:24:49 +02:00
Jim Martens b01cda512a Added some config values
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-10 15:23:30 +02:00
Jim Martens 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
Jim Martens 91b7febabb Changed train function to be an integration
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-10 12:15:17 +02:00
Jim Martens 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
Jim Martens 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
Jim Martens b926d11b84 Fixed path for train images
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-10 11:59:34 +02:00
Jim Martens db53c35929 Skip background boxes when saving training images
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-09 15:34:05 +02:00
Jim Martens 90fe1be4a6 Updated ssd_keras
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-09 14:21:57 +02:00
Jim Martens a0a976f9d7 Fixed saving debug images
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-09 14:21:38 +02:00
Jim Martens 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
Jim Martens c6e7dd3c50 Accounted for entropy in evaluation code
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-09 13:41:11 +02:00
Jim Martens 4cf782f7ab Fixed formatting
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-09 13:35:47 +02:00
Jim Martens b73cffb8c4 Removed empty line
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-09 13:35:18 +02:00
Jim Martens 7b3367739a Added evaluation path option to config
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-09 13:33:10 +02:00
Jim Martens 9069abcd57 Updated ssd_keras
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-09 13:30:04 +02:00
Jim Martens e25dfe86ca Updated ssd_keras
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-09 13:27:12 +02:00
Jim Martens 3418cb3861 Removed debug print
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-09 13:15:06 +02:00
Jim Martens 3fd7ba7373 Added conversion to numpy array
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-09 13:14:24 +02:00
Jim Martens c96783b384 Updated debug print
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-09 13:13:01 +02:00