Convert image input to BGR
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
@ -475,6 +475,7 @@ def load_scenenet_data(photo_paths: Sequence[Sequence[str]],
|
|||||||
else:
|
else:
|
||||||
transformations = [
|
transformations = [
|
||||||
object_detection_2d_photometric_ops.ConvertTo3Channels(),
|
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],
|
object_detection_2d_geometric_ops.Resize(height=resized_shape[0],
|
||||||
width=resized_shape[1])
|
width=resized_shape[1])
|
||||||
]
|
]
|
||||||
|
|||||||
Reference in New Issue
Block a user