67c098b0d6
Added conversion to int for class_id
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-07-10 16:01:18 +02:00
aaec64786c
Added debug print of class_id
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-07-10 15:55:01 +02:00
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
e6370ee92e
Fixed function call
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-07-10 15:46:08 +02:00
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
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
ed3fec54af
Added function to retrieve SSD model
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-07-10 15:24:49 +02:00
b01cda512a
Added some config values
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-07-10 15:23:30 +02:00
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
91b7febabb
Changed train function to be an integration
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-07-10 12:15:17 +02:00
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
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
b926d11b84
Fixed path for train images
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-07-10 11:59:34 +02:00
db53c35929
Skip background boxes when saving training images
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-07-09 15:34:05 +02:00
90fe1be4a6
Updated ssd_keras
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-07-09 14:21:57 +02:00
a0a976f9d7
Fixed saving debug images
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-07-09 14:21:38 +02:00
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
c6e7dd3c50
Accounted for entropy in evaluation code
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-07-09 13:41:11 +02:00
4cf782f7ab
Fixed formatting
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-07-09 13:35:47 +02:00
b73cffb8c4
Removed empty line
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-07-09 13:35:18 +02:00
7b3367739a
Added evaluation path option to config
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-07-09 13:33:10 +02:00
9069abcd57
Updated ssd_keras
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-07-09 13:30:04 +02:00
e25dfe86ca
Updated ssd_keras
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-07-09 13:27:12 +02:00
3418cb3861
Removed debug print
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-07-09 13:15:06 +02:00
3fd7ba7373
Added conversion to numpy array
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-07-09 13:14:24 +02:00
c96783b384
Updated debug print
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-07-09 13:13:01 +02:00
412f0ac453
Added debug print for predictions
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-07-09 13:00:23 +02:00
38ce65d367
Fixed not-compiled model
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-07-09 12:53:15 +02:00
9b13c5cf52
Fixed function call
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-07-09 12:25:11 +02:00
01a09ba474
Fixed SSD prediction
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-07-09 12:23:27 +02:00
311be36acf
Fixed requesting non-existent property
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-07-09 12:21:08 +02:00
b123d676bc
Predictor sizes only required for training
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-07-09 12:20:37 +02:00
4bdd069650
Added custom objects to recover saved model
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-07-09 12:13:41 +02:00
f5c8e9483a
Removed obsolete CLI options
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-07-09 12:05:23 +02:00
1d372aba59
Use one trajectory only for prediction
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-07-09 12:03:38 +02:00
0c7e96ee9d
Disabled early stopping
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-07-09 11:27:45 +02:00
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
94a4c69896
Changed paths to use config provided paths
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-07-07 13:55:01 +02:00
3e9c90f201
Added option "ssd_forward_passes_per_image"
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-07-07 13:50:59 +02:00
4886fda32e
Added missing argument for function call
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-07-07 13:43:19 +02:00
ecbd7ea00b
Fixed weights path
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-07-07 13:42:12 +02:00
322f2c51ab
Fixed save dir for models
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-07-07 13:39:45 +02:00
e3a05eb5fc
Truly implemented messaging for main module
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-07-04 19:39:43 +02:00
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
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
89adca3ac4
Removed obsolete import
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-07-04 17:11:29 +02:00
77e86f85e4
Added additional hyper parameters to config
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-07-04 17:11:13 +02:00
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
662c5ec7af
Extracted debug functionality into separate module
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-07-04 17:04:14 +02:00
eda3496f04
Updated the SSD training with new config
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-07-04 16:43:44 +02:00
