add norm method
Browse files
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):
|