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
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
cc9b3abe9b
Updated ssd_keras
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-13 14:04:42 +02:00
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
693594e75c
Improved function name for SceneNet data preparation
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-13 12:05:19 +02:00
58723adfe0
Corrected module doc
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-13 12:02:04 +02:00
a4e6ac2853
Added missing argument of num_epochs
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-11 12:06:22 +02:00
057bffcae3
Corrected summary path for ssd training
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-06-11 12:02:04 +02:00
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
