2martens/masterthesis
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity
625 Commits 1 Branch 0 Tags
8c31794aadc35826c5a0ccc542fbab352401e3fb
Commit Graph
609 Commits
Author SHA1 Message Date
2martens a66db745bf Convert example training image to numpy 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-06-24 10:56:28 +02:00
2martens 392927d4c7 Fixed saving of train image 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-06-24 10:52:52 +02:00
2martens a0b1c98fb4 Save example train image 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-06-24 10:41:51 +02:00
2martens 8678f14083 Update correct batch size 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-06-20 14:08:20 +02:00
2martens 78ab277ce7 Attempt batch size of 32 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-06-20 14:00:32 +02:00
2martens 4bc3cb2a81 Updated ssd_keras 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-06-20 13:45:55 +02:00
2martens 13f4abad0d Updated ssd_keras 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-06-20 13:40:47 +02:00
2martens 76ae44197e Updated ssd_keras 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-06-20 13:38:36 +02:00
2martens 5422bcb706 Updated ssd_keras 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-06-20 13:27:06 +02:00
2martens c22a98c833 Reversed previous commit 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-06-20 13:00:40 +02:00
2martens 0c02a94a33 Returned number of val batches to previous value 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-06-20 12:51:19 +02:00
2martens 0a2857f31d Updated ssd_keras 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-06-20 12:45:07 +02:00
2martens f14f0ae200 Set number of validation steps to 20 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-06-19 14:32:54 +02:00
2martens d1ee766579 Fixed save location for models and the weights 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-06-19 11:24:42 +02:00
2martens c6123607d9 Prevent remainder batch with fewer elements than rest 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-06-18 10:36:47 +02:00
2martens 9153828ff5 Remove images from data set that do not have instances 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-06-18 10:25:59 +02:00
2martens 9a5c96c46c Removed metrics that are unavailable in tensorflow 1.12 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-06-17 18:30:04 +02:00
2martens 2b64209bda Added missing function argument 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-06-17 18:20:02 +02:00
2martens b41b58930d Adapted evaluation code to new file structure of saved labels 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-06-13 17:20:50 +02:00
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