a4e6ac2853
Added missing argument of num_epochs
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-11 12:06:22 +02:00
057bffcae3
Corrected summary path for ssd training
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-11 12:02:04 +02:00
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
fe516f6429
Added additional values to print
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-11 11:34:18 +02:00
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
f6f6a88c50
Improve verbose printing for training
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-11 11:31:04 +02:00
e908248f89
Removed unnecessary empty lines
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-11 11:10:41 +02:00
6cebcb8177
Optimised verbose and debug console options
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-10 16:58:48 +02:00
4ccaad2ff0
Prevented division by zero errors
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-10 16:55:23 +02:00
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
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
15498549b7
Fixed mode of native SSD for val
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-10 14:17:50 +02:00
9dbfa5c640
Added train iteration argument to CLI
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-10 11:27:10 +02:00
e602e7339d
Provided checkpoint path in validation case
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-10 11:21:35 +02:00
4b85dd8376
Adjusted predict function to new checkpoint design
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-10 11:17:17 +02:00
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
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
1b52f4c9dc
Updated ssd_keras
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-05 12:48:28 +02:00
9f02297a52
Added steps to arguments for SSDInputEncoder
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-05 12:48:19 +02:00
efda35adb7
Fixed encoding of labels
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-05 12:35:08 +02:00
eadd4a9273
Added missing keyword argument
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-05 12:22:22 +02:00
fa2edbcede
Updated ssd_keras
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-05 12:19:30 +02:00
5397233b6e
Updated ssd_keras
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-05 12:09:55 +02:00
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
2ce6687efc
Updated ssd_keras
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-05 12:00:17 +02:00
3290bed0cb
Updated ssd_keras
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-05 11:57:08 +02:00
292e5e3c95
Implemented mising encoding for labels
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-05 11:49:44 +02:00
28abe6ae47
Ensure training only affects classifier layers
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-05 11:31:22 +02:00
5e7b16402b
Implemented SSD train action
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-05 11:09:46 +02:00
db9ab462ef
Extracted cli actions into separate module
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-05 10:57:46 +02:00
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
d370604279
Added another debug print
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-04 10:59:06 +02:00
810a390b4e
Added debug print
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-04 10:51:56 +02:00
9a32a170cf
Removed obsolete file
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-04 10:46:41 +02:00
80da94fcab
Improves wording of prepare functionality documentation
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-04 10:46:32 +02:00
13cecd2254
Improved formatting of main
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-04 10:36:18 +02:00
9851cef758
Improved formatting of train protobuf file
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-04 10:35:44 +02:00
2d5e3ff2e6
Added scenenet protobuf file for train data
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-04 10:27:41 +02:00
68b7377238
Updated ssd_keras
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-03 11:45:20 +02:00
9e0edb5d79
Fix not created directory
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-03 11:45:09 +02:00
5ea412a2d0
Ensure consistent arrays in match_predictions
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-03 11:43:22 +02:00
1c37dc8fb1
Added debug prints for get_precision_recall
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-03 11:39:27 +02:00
fb8a33cd24
Fixed indexing issue wrt the class_id
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-03 11:33:59 +02:00
dc6c046da2
Added debug print for class id
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-03 11:33:00 +02:00
2ba17a1540
Fixed cleaning of labels
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-03 11:32:01 +02:00
45d1a31129
Further debug print
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-03 11:27:39 +02:00
386ebd388c
Added debug print for label shape
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-03 11:26:02 +02:00
28a3111bde
Added debug print
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-03 11:19:33 +02:00
32eb98e9ff
Removed obsolete code
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-03 11:15:52 +02:00
970ed8d6c0
Convert labels to numpy array
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-03 11:14:50 +02:00
