How to give multiple arguments in tensorflow Model call function?

Question

I'm trying to build a model in tensorflow by extending the 'Model' class in tensorflow.keras. I need to pass two arguments in the 'call' function of this class, input images x (224,224,3) and output label y. But I get the following error while building the model:

ValueError: Currently, you cannot build your model if it has positional or keyword arguments that are not input to the model, but are required for its 'call' method.

class myCNN(Model):
  def __init__(self):
    super(myCNN, self).__init__()

    base_model = tf.keras.applications.VGG16(input_shape=(224,224,3), weights='imagenet')
    layer_name = 'block5_conv3'
    self.conv_1 = Model(inputs=base_model.input, outputs=base_model.get_layer(layer_name).output)
    self.flatten = L.Flatten(name='flatten')
    self.fc1 = L.Dense(1000, activation='relu', name='fc1')
    self.final = L.Activation('softmax')

  # The problem is because I need y
  def call(self, x, y):
    x = self.conv_1(x)
    x = self.flatten(x)
    x = self.fc1(x)
    return self.final(x)

model = myCNN()
model.build((None, 224, 224, 3, 1))

Why do you need y? You are not using it in the current computation inside call — Dr. Snoopy
– Dr. Snoopy, Commented Nov 6, 2019 at 8:33
I have not shown the usage. I need to mask x before passing it to flatten layer. The mask is dependant on y. As of now I did a workaround by passing [x,y] as a single argument to call() — Prerna Garg
– Prerna Garg, Commented Nov 6, 2019 at 11:21
That's not a workaround, it is how it should be implemented. — Dr. Snoopy
– Dr. Snoopy, Commented Nov 6, 2019 at 15:30

Shuhana Azmin · Accepted Answer · 2021-02-20 06:57:59Z

3

Inputs parameter of call method can be an input tensor or list/tuple of input tensors.

You can pass two arguments like this:

def call(self, inputs):
    x = inputs[0]
    y = inputs[1]
    x = self.conv_1(x)
    x = self.flatten(x)
    x = self.fc1(x)
    return self.final(x)

edited Feb 20, 2021 at 6:57

answered Feb 19, 2021 at 16:28

Shuhana Azmin

814 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

How to give multiple arguments in tensorflow Model call function?

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related