From 05d113a4ba00a9c7ce85d669e28127ee21c82b30 Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Wed, 7 Aug 2019 13:20:34 +0200 Subject: [PATCH] Convert image input to BGR Signed-off-by: Jim Martens --- src/twomartens/masterthesis/data.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/twomartens/masterthesis/data.py b/src/twomartens/masterthesis/data.py index 9c66bc3..2d15195 100644 --- 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]) ]