Jim Martens
|
ba2bd776ea
|
Improved formatting
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-11 12:22:10 +02:00 |
Jim Martens
|
0ae33130ad
|
Made image size parameter more compatible with clean code
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-11 12:22:01 +02:00 |
Jim Martens
|
9fa525ebe3
|
Added extra functions to compile model and get loss function
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-11 12:20:34 +02:00 |
Jim Martens
|
a848c12d64
|
Cleaned up ssd module and removed no longer needed code
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-11 11:24:06 +02:00 |
Jim Martens
|
253a507c90
|
Fixed indexing of instance array
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-10 17:36:33 +02:00 |
Jim Martens
|
b5058d9ee9
|
Updated debug print
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-10 17:33:15 +02:00 |
Jim Martens
|
82689ba6ec
|
Added debug print
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-10 17:31:02 +02:00 |
Jim Martens
|
ce87be6019
|
Added conversion to numpy array
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-10 17:29:35 +02:00 |
Jim Martens
|
25de799d80
|
Changed internal coordinates to corners
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-10 17:28:46 +02:00 |
Jim Martens
|
a685bba4fa
|
Properly reversed encoding of labels to visualize them
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-10 17:26:16 +02:00 |
Jim Martens
|
a97b9d8d21
|
Set internal coordinates to "corners"
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-10 17:12:15 +02:00 |
Jim Martens
|
4c4c05d383
|
Made it possible to save images without preventing training
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-10 16:29:39 +02:00 |
Jim Martens
|
d517c343be
|
Modified code to save images before and after encoding
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-10 16:16:08 +02:00 |
Jim Martens
|
c2d8da39c8
|
Fixed calculation of bbox coordinates
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-10 16:11:14 +02:00 |
Jim Martens
|
4f6611ec6c
|
Updated train_keras to work with SSD model directly
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-10 16:05:31 +02:00 |
Jim Martens
|
67c098b0d6
|
Added conversion to int for class_id
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-10 16:01:18 +02:00 |
Jim Martens
|
aaec64786c
|
Added debug print of class_id
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-10 15:55:01 +02:00 |
Jim Martens
|
7969716e6d
|
Made function compatible with encoded and decoded labels
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-10 15:51:20 +02:00 |
Jim Martens
|
e6370ee92e
|
Fixed function call
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-10 15:46:08 +02:00 |
Jim Martens
|
093cbe472e
|
Changed code to save debug images from processed but not encoded labels
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-10 15:38:06 +02:00 |
Jim Martens
|
21cef4a09e
|
Made ssd_train function compatible with clean code standards
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-10 15:25:18 +02:00 |
Jim Martens
|
ed3fec54af
|
Added function to retrieve SSD model
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-10 15:24:49 +02:00 |
Jim Martens
|
b01cda512a
|
Added some config values
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-10 15:23:30 +02:00 |
Jim Martens
|
d8d7b51655
|
Made load_scenenet_data more compatible with clean code principles
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-10 14:52:55 +02:00 |
Jim Martens
|
91b7febabb
|
Changed train function to be an integration
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-10 12:15:17 +02:00 |
Jim Martens
|
a90c6c7c92
|
Modified debug module to conform to clean code standards
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-10 12:12:13 +02:00 |
Jim Martens
|
4cf6723c28
|
Added ability to use custom string for debug save of train images
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-10 12:02:10 +02:00 |
Jim Martens
|
b926d11b84
|
Fixed path for train images
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-10 11:59:34 +02:00 |
Jim Martens
|
db53c35929
|
Skip background boxes when saving training images
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-09 15:34:05 +02:00 |
Jim Martens
|
90fe1be4a6
|
Updated ssd_keras
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-09 14:21:57 +02:00 |
Jim Martens
|
a0a976f9d7
|
Fixed saving debug images
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-09 14:21:38 +02:00 |
Jim Martens
|
bdd0862d68
|
Added guardian function to catch case with zero ground truth
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-09 13:45:01 +02:00 |
Jim Martens
|
c6e7dd3c50
|
Accounted for entropy in evaluation code
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-09 13:41:11 +02:00 |
Jim Martens
|
4cf782f7ab
|
Fixed formatting
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-09 13:35:47 +02:00 |
Jim Martens
|
b73cffb8c4
|
Removed empty line
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-09 13:35:18 +02:00 |
Jim Martens
|
7b3367739a
|
Added evaluation path option to config
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-09 13:33:10 +02:00 |
Jim Martens
|
9069abcd57
|
Updated ssd_keras
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-09 13:30:04 +02:00 |
Jim Martens
|
e25dfe86ca
|
Updated ssd_keras
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-09 13:27:12 +02:00 |
Jim Martens
|
3418cb3861
|
Removed debug print
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-09 13:15:06 +02:00 |
Jim Martens
|
3fd7ba7373
|
Added conversion to numpy array
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-09 13:14:24 +02:00 |
Jim Martens
|
c96783b384
|
Updated debug print
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-09 13:13:01 +02:00 |
Jim Martens
|
412f0ac453
|
Added debug print for predictions
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-09 13:00:23 +02:00 |
Jim Martens
|
38ce65d367
|
Fixed not-compiled model
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-09 12:53:15 +02:00 |
Jim Martens
|
9b13c5cf52
|
Fixed function call
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-09 12:25:11 +02:00 |
Jim Martens
|
01a09ba474
|
Fixed SSD prediction
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-09 12:23:27 +02:00 |
Jim Martens
|
311be36acf
|
Fixed requesting non-existent property
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-09 12:21:08 +02:00 |
Jim Martens
|
b123d676bc
|
Predictor sizes only required for training
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-09 12:20:37 +02:00 |
Jim Martens
|
4bdd069650
|
Added custom objects to recover saved model
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-09 12:13:41 +02:00 |
Jim Martens
|
f5c8e9483a
|
Removed obsolete CLI options
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-09 12:05:23 +02:00 |
Jim Martens
|
1d372aba59
|
Use one trajectory only for prediction
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-09 12:03:38 +02:00 |