2martens/masterthesis
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity
377 Commits 1 Branch 0 Tags
9dbfa5c640d63400fc0ac4144df91975b1512c7e
Commit Graph

377 Commits

This Branch
This Branch
All Branches
Author SHA1 Message Date
Jim Martens
9dbfa5c640 Added train iteration argument to CLI 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-06-10 11:27:10 +02:00
Jim Martens
e602e7339d Provided checkpoint path in validation case 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-06-10 11:21:35 +02:00
Jim Martens
4b85dd8376 Adjusted predict function to new checkpoint design 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-06-10 11:17:17 +02:00
Jim Martens
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
Jim Martens
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
Jim Martens
1b52f4c9dc Updated ssd_keras 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-06-05 12:48:28 +02:00
Jim Martens
9f02297a52 Added steps to arguments for SSDInputEncoder 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-06-05 12:48:19 +02:00
Jim Martens
efda35adb7 Fixed encoding of labels 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-06-05 12:35:08 +02:00
Jim Martens
eadd4a9273 Added missing keyword argument 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-06-05 12:22:22 +02:00
Jim Martens
fa2edbcede Updated ssd_keras 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-06-05 12:19:30 +02:00
Jim Martens
5397233b6e Updated ssd_keras 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-06-05 12:09:55 +02:00
Jim Martens
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
Jim Martens
2ce6687efc Updated ssd_keras 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-06-05 12:00:17 +02:00
Jim Martens
3290bed0cb Updated ssd_keras 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-06-05 11:57:08 +02:00
Jim Martens
292e5e3c95 Implemented mising encoding for labels 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-06-05 11:49:44 +02:00
Jim Martens
28abe6ae47 Ensure training only affects classifier layers 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-06-05 11:31:22 +02:00
Jim Martens
5e7b16402b Implemented SSD train action 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-06-05 11:09:46 +02:00
Jim Martens
db9ab462ef Extracted cli actions into separate module 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-06-05 10:57:46 +02:00
Jim Martens
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
Jim Martens
d370604279 Added another debug print 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-06-04 10:59:06 +02:00
Jim Martens
810a390b4e Added debug print 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-06-04 10:51:56 +02:00
Jim Martens
9a32a170cf Removed obsolete file 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-06-04 10:46:41 +02:00
Jim Martens
80da94fcab Improves wording of prepare functionality documentation 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-06-04 10:46:32 +02:00
Jim Martens
13cecd2254 Improved formatting of main 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-06-04 10:36:18 +02:00
Jim Martens
9851cef758 Improved formatting of train protobuf file 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-06-04 10:35:44 +02:00
Jim Martens
2d5e3ff2e6 Added scenenet protobuf file for train data 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-06-04 10:27:41 +02:00
Jim Martens
68b7377238 Updated ssd_keras 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-06-03 11:45:20 +02:00
Jim Martens
9e0edb5d79 Fix not created directory 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-06-03 11:45:09 +02:00
Jim Martens
5ea412a2d0 Ensure consistent arrays in match_predictions 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-06-03 11:43:22 +02:00
Jim Martens
1c37dc8fb1 Added debug prints for get_precision_recall 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-06-03 11:39:27 +02:00
Jim Martens
fb8a33cd24 Fixed indexing issue wrt the class_id 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-06-03 11:33:59 +02:00
Jim Martens
dc6c046da2 Added debug print for class id 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-06-03 11:33:00 +02:00
Jim Martens
2ba17a1540 Fixed cleaning of labels 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-06-03 11:32:01 +02:00
Jim Martens
45d1a31129 Further debug print 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-06-03 11:27:39 +02:00
Jim Martens
386ebd388c Added debug print for label shape 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-06-03 11:26:02 +02:00
Jim Martens
28a3111bde Added debug print 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-06-03 11:19:33 +02:00
Jim Martens
32eb98e9ff Removed obsolete code 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-06-03 11:15:52 +02:00
Jim Martens
970ed8d6c0 Convert labels to numpy array 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-06-03 11:14:50 +02:00
Jim Martens
8a239128d5 Attempt to fix exclusion of padded label entries 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-06-03 10:52:21 +02:00
Jim Martens
e097643daf Fixed wrong indexing 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-06-03 10:48:58 +02:00
Jim Martens
71c1b1f197 Background boxes are no longer removed before building observations 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-06-03 10:38:44 +02:00
Jim Martens
c0e5f74bee Reduced forward passes to 10 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-06-03 10:37:08 +02:00
Jim Martens
145b929863 Removed debug print 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-06-03 10:36:37 +02:00
Jim Martens
9c3e436579 Prints non-zero classes 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-06-03 10:32:26 +02:00
Jim Martens
225d0307eb Added code to debug probability of background class 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-05-30 14:59:52 +02:00
Jim Martens
bfc4adc93e Added debug print for class ids 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-05-30 14:52:20 +02:00
Jim Martens
91ebf9032c Provided image height and width for decode detections 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-05-30 14:49:53 +02:00
Jim Martens
3d1ba8adbf Removed debug prints during forward passes 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-05-30 14:46:55 +02:00
Jim Martens
a838dd6a1b Fixed elimination of background boxes 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-05-30 14:46:18 +02:00
Jim Martens
9544d5d3df Removed background detections before iou for observations 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-05-30 14:42:16 +02:00