8828f67d5d41b88881bd0d7d17a9dbc9a37f98b7,models/dist_model.py,DistModel,forward,#DistModel#Any#Any#Any#,74

Before Change

        if self.spatial:
            L = [convert_output(x) for x in self.d0]
            height = functools.reduce(lcm, [x.shape[0] for x in L])
            width  = functools.reduce(lcm, [x.shape[1] for x in L])
            L = [np.repeat(np.repeat(x, height//x.shape[0], 0), width//x.shape[1], 1) for x in L]
            L = np.mean(np.concatenate(L, 2) * len(L), 2)
            return L
        else:

After Change

            L = [convert_output(x) for x in self.d0]
            spatial_shape = self.spatial_shape
            if spatial_shape is None:
                spatial_shape = (max([x.shape[0] for x in L])*self.spatial_factor, max([x.shape[1] for x in L])*self.spatial_factor)
            
            L = [skimage.transform.resize(x, spatial_shape, order=self.spatial_order, mode="edge") for x in L]
            
            L = np.mean(np.concatenate(L, 2) * len(L), 2)

Instances