Hi, I’m trying to implement SOLO architecture for instance segmentation in TensorFlow (Decoupled version).
https://arxiv.org/pdf/1912.04488.pdf
Right now, I need to compute the loss function and multiply each output map from conv2d to each other.
xi = Conv2D(…)(input) # output is (batch, None, None, 24)
yi = Conv2D(…)(input) # output is (batch, None, None, 24)
I need to multiply output filters (element wise) xi with yi in a way to get output with (batch, None, None, 24*24).
I try to do this with for cycles but get error “OperatorNotAllowedInGraphError: iterating over tf.Tensor is not allowed: AutoGraph did convert this function”.
Any advice to achieve this?