Commit Graph

722 Commits

Author SHA1 Message Date
01f5de6d6c Changed config get function to return correct type
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-04 16:33:59 +02:00
a2fe3e860a Added parameters for batch size and ssd image size
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-04 16:30:29 +02:00
b6beebd0c1 Updated CLI arguments for prepare action
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-04 16:27:43 +02:00
3354a033fe Updated prepare action to use new config
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-04 16:25:57 +02:00
f6c66d1d34 Reordered functions in cli module
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-04 16:15:50 +02:00
55b0e7d6b5 Added missing printing of retrieved option value
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-04 15:56:42 +02:00
75a5d9458b Config file is now updated after changing config property
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-04 15:55:55 +02:00
e805034b58 Improved list action to actually print config values
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-04 15:51:31 +02:00
7819f6fceb Fixed problem with non-str values for config parser
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-04 15:48:14 +02:00
2fe02452ea Removed duplicate matplotlib requirement
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-04 15:42:16 +02:00
5e089c4bfc Improved help text for debug option
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-04 15:37:16 +02:00
bf7f36174f Implemented config functionality
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-04 15:36:29 +02:00
f32ff0b9f0 Added config actions with skeleton implementation
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-04 14:39:14 +02:00
34a23597b5 Renamed "action" sub parsers on global level to "component"
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-04 14:28:11 +02:00
136b52a7f4 Added config to CLI parsers
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-04 14:25:58 +02:00
bd499425ed Only use one trajectory for training and validation data each
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-04 13:15:51 +02:00
7f5a32a441 Added ability limit number of trajectories considered
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-04 13:15:33 +02:00
d3be3ea5f9 Pickle history property instead of History object
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-04 13:02:06 +02:00
2b93480df9 Reversed optimizer to Tensorflow and changed beta1 to 0.9
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-04 12:59:51 +02:00
59a1d85af9 Change to Keras optimizer
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-04 12:33:34 +02:00
d9b742f61a Improve argmax usage
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-02 16:00:58 +02:00
4595fa85d4 Removed empty line
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-02 15:56:08 +02:00
2e609e5725 Removed debug print
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-02 15:51:01 +02:00
a0a4193dea Remove obsolete if statement
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-02 15:49:33 +02:00
76de6e34fc Fixed missing directory
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-02 15:48:12 +02:00
22aa463152 Added code to save training images with bounding boxes
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-02 15:45:04 +02:00
112dc48f36 Disable augmentation of input for now
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-02 15:10:40 +02:00
e4e94940b8 Implemented task to measure number of instances per COCO class per trajectory
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-02 14:20:50 +02:00
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