feat: application of normalization for 3-channel inputs.
@@ -402,7 +402,11 @@ def ConvNeXt(depths,
inputs = img_input
x = inputs
+
if include_preprocessing:
+ channel_axis = 3 if backend.image_data_format() == "channels_last" else 1
+ num_channels = input_shape[channel_axis - 1]
+ if num_channels == 3:
x = PreStem(name=model_name)(x)
# Stem block.
