Commit Graph

694 Commits

Author SHA1 Message Date
6cf8f515d4 Updated ssd_keras
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-02 14:17:02 +02:00
0d14808ba7 Added progress bar to data preparation
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-02 11:59:54 +02:00
15c83940fe Added wordnet id to visualised bboxes
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-02 11:53:40 +02:00
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
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
3d7d97355c Reverse indices to what it should be
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-02 11:44:43 +02:00
7ad1789e24 Fixed wrong axis in preparation code
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-02 11:43:56 +02:00
f72ef5218a Attempt to switch axes
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-02 11:38:43 +02:00
da4e624f41 Close figure after usage
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-02 11:29:54 +02:00
1191761149 Fixed accessing of colour list
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-02 11:28:49 +02:00
f2c718540a Fixed label retrieval
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-02 11:27:02 +02:00
dd619179c5 Fixed classes variable assignment
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-02 11:25:01 +02:00
58d85a7efc Add debug print
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-02 11:21:26 +02:00
eb9f1fb3a2 Fixed typo
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-02 11:18:17 +02:00
360f84f954 Added visualise task
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-02 11:13:38 +02:00
b79ba0bea4 Updated ssd_keras
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-02 10:34:17 +02:00
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
180556f6ba Correctly access one training image
Signed-off-by: Jim Martens <github@2martens.de>
2019-06-24 11:04:57 +02:00
25c4908825 Fixed retrieval of train image
Signed-off-by: Jim Martens <github@2martens.de>
2019-06-24 11:00:51 +02:00
ce6d67cf4c Added debug print for example train image
Signed-off-by: Jim Martens <github@2martens.de>
2019-06-24 10:57:57 +02:00
a66db745bf Convert example training image to numpy
Signed-off-by: Jim Martens <github@2martens.de>
2019-06-24 10:56:28 +02:00
392927d4c7 Fixed saving of train image
Signed-off-by: Jim Martens <github@2martens.de>
2019-06-24 10:52:52 +02:00
a0b1c98fb4 Save example train image
Signed-off-by: Jim Martens <github@2martens.de>
2019-06-24 10:41:51 +02:00
8678f14083 Update correct batch size
Signed-off-by: Jim Martens <github@2martens.de>
2019-06-20 14:08:20 +02:00
78ab277ce7 Attempt batch size of 32
Signed-off-by: Jim Martens <github@2martens.de>
2019-06-20 14:00:32 +02:00
4bc3cb2a81 Updated ssd_keras
Signed-off-by: Jim Martens <github@2martens.de>
2019-06-20 13:45:55 +02:00
13f4abad0d Updated ssd_keras
Signed-off-by: Jim Martens <github@2martens.de>
2019-06-20 13:40:47 +02:00
76ae44197e Updated ssd_keras
Signed-off-by: Jim Martens <github@2martens.de>
2019-06-20 13:38:36 +02:00
5422bcb706 Updated ssd_keras
Signed-off-by: Jim Martens <github@2martens.de>
2019-06-20 13:27:06 +02:00
c22a98c833 Reversed previous commit
Signed-off-by: Jim Martens <github@2martens.de>
2019-06-20 13:00:40 +02:00
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
0a2857f31d Updated ssd_keras
Signed-off-by: Jim Martens <github@2martens.de>
2019-06-20 12:45:07 +02:00
e204ef6617 Added imutils to required packages
Signed-off-by: Jim Martens <github@2martens.de>
2019-06-20 12:44:52 +02:00
f14f0ae200 Set number of validation steps to 20
Signed-off-by: Jim Martens <github@2martens.de>
2019-06-19 14:32:54 +02:00
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
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
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
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
2b64209bda Added missing function argument
Signed-off-by: Jim Martens <github@2martens.de>
2019-06-17 18:20:02 +02:00
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
232e531a9d Changed name of val to test
Signed-off-by: Jim Martens <github@2martens.de>
2019-06-13 17:03:29 +02:00
f9da5e8236 Improved name of test to evaluate
Signed-off-by: Jim Martens <github@2martens.de>
2019-06-13 17:01:00 +02:00
b2a1db6485 Updated predict functionality to use Keras
Signed-off-by: Jim Martens <github@2martens.de>
2019-06-13 16:56:12 +02:00
d466e8e376 Added predict function that uses Keras functionality
Signed-off-by: Jim Martens <github@2martens.de>
2019-06-13 16:52:25 +02:00
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
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
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
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
c260b7d824 Finished conversion of training functionality to keras
Signed-off-by: Jim Martens <github@2martens.de>
2019-06-13 15:07:46 +02:00
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