1c6277ecb801327ce4b2e376e1e13a18b79ba52c,neuron/layers.py,DenseSpatialTransformer,call,#DenseSpatialTransformer#Any#,129

Before Change

        // check shapes
        assert len(inputs) == 2, "inputs has to be len 2, found: %d" % len(inputs)
        if inputs[1].shape[1:-1] != inputs[0].shape[1:-1]:
            raise Exception("Shift shape should match vol shape. "
                            "Got: " + inputs[1].shape[1:-1] + " and " + inputs[0].shape[1:-1])

        // prepare location shift
        loc_shift = inputs[1]
        if self.indexing == "xy":  // shift the first two dimensions
            loc_shift_split = tf.split(loc_shift, loc_shift.shape[-1], axis=-1)
            loc_shift_lst = [loc_shift_split[1], loc_shift_split[0], *loc_shift_split[2:]]

After Change

        if not self.is_affine:
            if not all([trf.shape[1:-1][f] != vol.shape[1:-1][f] for f in range(self.ndims)]):
                raise Exception("Shift shape should match vol shape. "
                                "Got: " + str(trf.shape[1:-1]) + " and " + str(vol.shape[1:-1]))

        // go from affine
        if self.is_affine:

Instances