pranay-ar commited on
Commit
a6d9ff5
·
1 Parent(s): 44547fd

add norm method

Browse files
Files changed (1) hide show
  1. image_processing_minicpmv.py +2 -2
image_processing_minicpmv.py CHANGED
@@ -124,8 +124,8 @@ class MiniCPMVImageProcessor(BaseImageProcessor):
124
  self.im_id_start = kwargs.pop("im_id_start", "<image_id>")
125
  self.im_id_end = kwargs.pop("im_id_end", "</image_id>")
126
  self.slice_mode = kwargs.pop("slice_mode", True)
127
- self.mean = np.array(kwargs.pop("norm_mean", [0.5, 0.5, 0.5]))
128
- self.std = np.array(kwargs.pop("norm_std", [0.5, 0.5, 0.5]))
129
  self.version = kwargs.pop("version", 2.0)
130
 
131
  def ensure_divide(self, length, patch_size):
 
124
  self.im_id_start = kwargs.pop("im_id_start", "<image_id>")
125
  self.im_id_end = kwargs.pop("im_id_end", "</image_id>")
126
  self.slice_mode = kwargs.pop("slice_mode", True)
127
+ self.mean = np.array(kwargs.pop("norm_mean", [0.5, 0.5, 0.5])).flatten()
128
+ self.std = np.array(kwargs.pop("norm_std", [0.5, 0.5, 0.5])).flatten()
129
  self.version = kwargs.pop("version", 2.0)
130
 
131
  def ensure_divide(self, length, patch_size):