2martens
232e531a9d
Changed name of val to test
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-13 17:03:29 +02:00
2martens
f9da5e8236
Improved name of test to evaluate
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-13 17:01:00 +02:00
2martens
b2a1db6485
Updated predict functionality to use Keras
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-13 16:56:12 +02:00
2martens
d466e8e376
Added predict function that uses Keras functionality
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-13 16:52:25 +02:00
2martens
77a195a144
Improved data generation to cover evaluation cases as well
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-13 16:51:59 +02:00
2martens
f8bed423e4
Added inverse transforms to output of non-training generators
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-13 16:06:21 +02:00
2martens
cc74a8674e
Added missing steps and aspect ratios to the label encoder
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-13 15:58:44 +02:00
2martens
49a320ca12
Changed mode argument to training argument to better reflect requirements
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-13 15:12:37 +02:00
2martens
c260b7d824
Finished conversion of training functionality to keras
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-13 15:07:46 +02:00
2martens
7d287c4432
Added train function which utilises the Keras train functions
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-13 15:00:54 +02:00
2martens
88010c5914
Added type hinting for attributes of SSD wrapper objects
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-13 14:12:13 +02:00
2martens
cc9b3abe9b
Updated ssd_keras
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-13 14:04:42 +02:00
2martens
87133dd885
Modified data module to work with SSD data generator
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-13 13:30:02 +02:00
2martens
693594e75c
Improved function name for SceneNet data preparation
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-13 12:05:19 +02:00
2martens
58723adfe0
Corrected module doc
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-13 12:02:04 +02:00
2martens
a4e6ac2853
Added missing argument of num_epochs
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-11 12:06:22 +02:00
2martens
057bffcae3
Corrected summary path for ssd training
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-11 12:02:04 +02:00
2martens
eef6078533
Added print to show epoch number
...
This can be used to identify if the data set is in fact repeated.
In that case this print line has to be printed at least twice.
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-11 12:00:08 +02:00
2martens
fe516f6429
Added additional values to print
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-11 11:34:18 +02:00
2martens
83b9e670f8
Moved loss metric assignment into begin of epoch
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-11 11:34:04 +02:00
2martens
f6f6a88c50
Improve verbose printing for training
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-11 11:31:04 +02:00
2martens
e908248f89
Removed unnecessary empty lines
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-11 11:10:41 +02:00
2martens
6cebcb8177
Optimised verbose and debug console options
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-10 16:58:48 +02:00
2martens
4ccaad2ff0
Prevented division by zero errors
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-10 16:55:23 +02:00
2martens
ca90ea9f17
Fixed guardian if to work with numpy arrays
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-10 16:55:07 +02:00
2martens
488703bb26
Made f1 score calculation robust against classes with zero predictions
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-10 16:49:17 +02:00
2martens
15498549b7
Fixed mode of native SSD for val
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-10 14:17:50 +02:00
2martens
9dbfa5c640
Added train iteration argument to CLI
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-10 11:27:10 +02:00
2martens
e602e7339d
Provided checkpoint path in validation case
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-10 11:21:35 +02:00
2martens
4b85dd8376
Adjusted predict function to new checkpoint design
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-10 11:17:17 +02:00
2martens
110e098d78
Removed double-iteration over epochs
...
1. Dataset was repeated nr_epochs times
2. loop over dataset was repeated nr_epoch times
Consequence: each checkpointed epoch was in fact nr_epochs long
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-06 11:06:59 +02:00
2martens
c893954120
Added aspect_ratios_per_layer to arguments for SSDInputEncoder
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-05 12:56:07 +02:00
2martens
1b52f4c9dc
Updated ssd_keras
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-05 12:48:28 +02:00
2martens
9f02297a52
Added steps to arguments for SSDInputEncoder
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-05 12:48:19 +02:00
2martens
efda35adb7
Fixed encoding of labels
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-05 12:35:08 +02:00
2martens
eadd4a9273
Added missing keyword argument
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-05 12:22:22 +02:00
2martens
fa2edbcede
Updated ssd_keras
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-05 12:19:30 +02:00
2martens
5397233b6e
Updated ssd_keras
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-05 12:09:55 +02:00
2martens
a586dd8297
Fix issue with non-checkpointable custom class object
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-05 12:06:05 +02:00
2martens
2ce6687efc
Updated ssd_keras
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-05 12:00:17 +02:00
2martens
3290bed0cb
Updated ssd_keras
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-05 11:57:08 +02:00
2martens
292e5e3c95
Implemented mising encoding for labels
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-05 11:49:44 +02:00
2martens
28abe6ae47
Ensure training only affects classifier layers
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-05 11:31:22 +02:00
2martens
5e7b16402b
Implemented SSD train action
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-05 11:09:46 +02:00
2martens
db9ab462ef
Extracted cli actions into separate module
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-05 10:57:46 +02:00
2martens
f1fe8ddfc5
Fixed if clause to catch empty list as well
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-04 11:00:25 +02:00
2martens
d370604279
Added another debug print
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-04 10:59:06 +02:00
2martens
810a390b4e
Added debug print
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-04 10:51:56 +02:00
2martens
9a32a170cf
Removed obsolete file
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-04 10:46:41 +02:00
2martens
80da94fcab
Improves wording of prepare functionality documentation
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-04 10:46:32 +02:00