from keras.layers import Conv2D
from keras.models import Sequential


def srcnn(input_shape, c=1, f1=9, f2=1, f3=5, n1=64, n2=32):
    model = Sequential()
    model.add(Conv2D(nb_filter=n1, nb_row=f1, nb_col=f1,
                     activation='relu', init='he_normal',
                     input_shape=input_shape))
    model.add(Conv2D(nb_filter=n2, nb_row=f2, nb_col=f2,
                     activation='relu', init='he_normal'))
    model.add(Conv2D(nb_filter=c, nb_row=f3, nb_col=f3))
    return model

from keras.layers import Conv2D
from keras.models import Sequential


def srcnn(input_shape, c=1, f1=9, f2=1, f3=5, n1=64, n2=32):
    model = Sequential()
    model.add(Conv2D(nb_filter=n1, nb_row=f1, nb_col=f1, init='he_normal',
                     activation='relu', input_shape=input_shape))
    model.add(Conv2D(nb_filter=n2, nb_row=f2, nb_col=f2, init='he_normal',
                     activation='relu'))
    model.add(Conv2D(nb_filter=c, nb_row=f3, nb_col=f3, init='he_normal'))
    return model

Use he_normal initialization for all layers
Use he_normal initialization for all layers

  from keras.layers import Conv2D
  from keras.models import Sequential
  
  
  def srcnn(input_shape, c=1, f1=9, f2=1, f3=5, n1=64, n2=32):
      model = Sequential()
-     model.add(Conv2D(nb_filter=n1, nb_row=f1, nb_col=f1,
+     model.add(Conv2D(nb_filter=n1, nb_row=f1, nb_col=f1, init='he_normal',
-                      activation='relu', init='he_normal',
-                      input_shape=input_shape))
+                      activation='relu', input_shape=input_shape))
-     model.add(Conv2D(nb_filter=n2, nb_row=f2, nb_col=f2,
+     model.add(Conv2D(nb_filter=n2, nb_row=f2, nb_col=f2, init='he_normal',
-                      activation='relu', init='he_normal'))
+                      activation='relu'))
-     model.add(Conv2D(nb_filter=c, nb_row=f3, nb_col=f3))
+     model.add(Conv2D(nb_filter=c, nb_row=f3, nb_col=f3, init='he_normal'))
      return model
  
Use he_normal initialization for all layers
## Code Before:
from keras.layers import Conv2D
from keras.models import Sequential


def srcnn(input_shape, c=1, f1=9, f2=1, f3=5, n1=64, n2=32):
    model = Sequential()
    model.add(Conv2D(nb_filter=n1, nb_row=f1, nb_col=f1,
                     activation='relu', init='he_normal',
                     input_shape=input_shape))
    model.add(Conv2D(nb_filter=n2, nb_row=f2, nb_col=f2,
                     activation='relu', init='he_normal'))
    model.add(Conv2D(nb_filter=c, nb_row=f3, nb_col=f3))
    return model

## Instruction:
Use he_normal initialization for all layers
## Code After:
from keras.layers import Conv2D
from keras.models import Sequential


def srcnn(input_shape, c=1, f1=9, f2=1, f3=5, n1=64, n2=32):
    model = Sequential()
    model.add(Conv2D(nb_filter=n1, nb_row=f1, nb_col=f1, init='he_normal',
                     activation='relu', input_shape=input_shape))
    model.add(Conv2D(nb_filter=n2, nb_row=f2, nb_col=f2, init='he_normal',
                     activation='relu'))
    model.add(Conv2D(nb_filter=c, nb_row=f3, nb_col=f3, init='he_normal'))
    return model
