ceedc39e529dbd8a66bdb3950af131b75b430683,tensorflow_/tensorflowcv/models/vgg.py,VGG,__call__,#VGG#Any#Any#,242

Before Change

                name="features/stage{}/pool".format(i + 1))

        in_channels = in_channels * 7 * 7
        x = tf.reshape(x, [-1, in_channels])
        x = vgg_output_block(
            x=x,
            in_channels=in_channels,
            classes=self.classes,

After Change

        in_channels = in_channels * 7 * 7
        // x = tf.reshape(x, [-1, in_channels])
        x = flatten(
            x=x,
            out_channels=in_channels,
            data_format=self.data_format)
        x = vgg_output_block(
            x=x,
            in_channels=in_channels,
            classes=self.classes,

Instances