Convert image input to BGR

Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens 2019-08-07 13:20:34 +02:00
parent 29522921f1
commit 05d113a4ba
1 changed files with 1 additions and 0 deletions

View File

@ -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])
]