Commit Graph
441 Commits
Author SHA1 Message Date
2martens 92d60ad4e3 Added wordnet class name to visualised ground truth bboxes
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-02 11:52:17 +02:00
2martens 8bddf320f8 Added wordnet class name to prepared ground truth
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-02 11:51:23 +02:00
2martens 3d7d97355c Reverse indices to what it should be
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-02 11:44:43 +02:00
2martens 7ad1789e24 Fixed wrong axis in preparation code
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-02 11:43:56 +02:00
2martens f72ef5218a Attempt to switch axes
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-02 11:38:43 +02:00
2martens da4e624f41 Close figure after usage
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-02 11:29:54 +02:00
2martens 1191761149 Fixed accessing of colour list
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-02 11:28:49 +02:00
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