diff --git a/externals/ABELE/ilore/ineighgen.py b/externals/ABELE/ilore/ineighgen.py index fd0cc00..bf2fe48 100755 --- a/externals/ABELE/ilore/ineighgen.py +++ b/externals/ABELE/ilore/ineighgen.py @@ -240,8 +240,7 @@ def generate(self, img, num_samples=1000): # print(Yb[:5]) # lZ_img, Z_img, Yb = self._fix_neigh(lZ_img, Z_img, Yb, class_value) - - Z = np.array(lZ_img) + if self.scale: scaler = MinMaxScaler() Z = scaler.fit_transform(Z)