Convert image input to BGR

Signed-off-by: Jim Martens <github@2martens.de>
2019-08-07 13:20:34 +02:00
parent 29522921f1
commit 05d113a4ba
1 changed files with 1 additions and 0 deletions
--- a/src/twomartens/masterthesis/data.py
+++ b/src/twomartens/masterthesis/data.py
@ -475,6 +475,7 @@ def load_scenenet_data(photo_paths: Sequence[Sequence[str]],
    else:
        transformations = [
            object_detection_2d_photometric_ops.ConvertTo3Channels(),
+            object_detection_2d_photometric_ops.ConvertColor(current='RGB', to='BGR'),
            object_detection_2d_geometric_ops.Resize(height=resized_shape[0],
                                                     width=resized_shape[1])
        ]