Commit Graph
734 Commits
Author SHA1 Message Date
2martens f2c718540a Fixed label retrieval
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-02 11:27:02 +02:00
2martens dd619179c5 Fixed classes variable assignment
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-02 11:25:01 +02:00
2martens 58d85a7efc Add debug print
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-02 11:21:26 +02:00
2martens eb9f1fb3a2 Fixed typo
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-02 11:18:17 +02:00
2martens 360f84f954 Added visualise task
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-02 11:13:38 +02:00
2martens b79ba0bea4 Updated ssd_keras
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-02 10:34:17 +02:00
2martens 93c8643373 Entire first batch of train images are saved
Signed-off-by: Jim Martens <github@2martens.de>
2019-06-24 11:10:25 +02:00
2martens 180556f6ba Correctly access one training image
Signed-off-by: Jim Martens <github@2martens.de>
2019-06-24 11:04:57 +02:00
2martens 25c4908825 Fixed retrieval of train image
Signed-off-by: Jim Martens <github@2martens.de>
2019-06-24 11:00:51 +02:00
2martens ce6d67cf4c Added debug print for example train image
Signed-off-by: Jim Martens <github@2martens.de>
2019-06-24 10:57:57 +02:00
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 e204ef6617 Added imutils to required packages
Signed-off-by: Jim Martens <github@2martens.de>
2019-06-20 12:44:52 +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