5b522a3b1952bd1b531ceffa07138639b936f030,hypergan/generators/segment_generator.py,SegmentGenerator,build,#SegmentGenerator#Any#,14

Before Change

        self.ops.add_weights(mask_generator.variables())
        self.ops.add_weights(g1.variables())
        self.ops.add_weights(g2.variables())

        sample = (g1.sample * mask) + \
                      (1.0-mask) * g2.sample 

After Change

        print("OUTPUT", sample, g1.sample, g2.sample, mask)

        if not hasattr(self, "g1"):
            self.ops.add_weights(mask_generator.variables())
            self.ops.add_weights(g1.variables())
            self.ops.add_weights(g2.variables())


            self.g1 = g1
            self.g2 = g2

            self.g1x = (g1.sample * mask) + \
                    (1.0-mask) * gan.inputs.x
            self.g2x = (gan.inputs.x * mask) + \
                    (1.0-mask) * g2.sample

            self.mask_generator = mask_generator
            self.mask = tf.tile(mask_generator.sample, [1,1,1,3])
            if config.mask_generator:
                self.mask = self.mask * 2 - 1

        else:
            print("-->SKIP")


        return sample

Instances