738c8758e30c6f8bdf9155529aeb431547af65d6,python/caffe/pycaffe.py,,_Net_set_mean,#,184

Before Change

        raise Exception("Input not in {}".format(self.inputs))
    mean = np.load(mean_f)
    if mode == "elementwise":
        if mean.shape != self.input.data.shape[1:]:
            raise Exception("The mean shape does not match the input shape.")
        self.mean[input_] = mean
    elif mode == "channel":
        self.mean[input_] = mean.mean(1).mean(1)
    else:

After Change

    in_shape = self.blobs[input_].data.shape
    mean = np.load(mean_f)
    if mode == "elementwise":
        if mean.shape != in_shape[1:]:
            mean = caffe.io.resize_image(mean.transpose((1,2,0)),
                    in_shape[2:]).transpose((2,0,1))
        self.mean[input_] = mean
    elif mode == "channel":
        self.mean[input_] = mean.mean(1).mean(1).reshape((in_shape[1], 1, 1))
    else:

Instances