diff transformations/thick.py @ 26:47e7202d4f19

Array format handling bug fix for thick.py
author Xavier Glorot <glorotxa@iro.umontreal.ca>
date Fri, 29 Jan 2010 14:09:14 -0500
parents 442789c94b27
children 5e00ed18ae32
line wrap: on
line diff
--- a/transformations/thick.py	Fri Jan 29 11:44:12 2010 -0500
+++ b/transformations/thick.py	Fri Jan 29 14:09:14 2010 -0500
@@ -69,15 +69,14 @@
             maxi = float(N.max(image))
             mini = float(N.min(image))
             
-            if maxi>1.0:
-                image=image/maxi
+            imagenorm=image/maxi
             
             if self.meth==1:
                 trans=scipy.ndimage.morphology.grey_dilation\
-                    (image,size=self.__structuring_elements__[self.nb].shape,structure=self.__structuring_elements__[self.nb])
+                    (imagenorm,size=self.__structuring_elements__[self.nb].shape,structure=self.__structuring_elements__[self.nb])
             else:
                 trans=scipy.ndimage.morphology.grey_erosion\
-                    (image,size=self.__structuring_elements__[self.nb].shape,structure=self.__structuring_elements__[self.nb])
+                    (imagenorm,size=self.__structuring_elements__[self.nb].shape,structure=self.__structuring_elements__[self.nb])
             
             #------renormalizing
             maxit = N.max(trans)
@@ -92,9 +91,10 @@
         if image.shape == (self.__nx__,self.__ny__):
             return self.transform_1_image(image)
         if image.ndim == 3:
+            newimage = copy.copy(image)
             for i in range(image.shape[0]):
-                image[i,:,:] = self.transform_1_image(image[i,:,:])
-            return N.reshape(newimage,image.shape)
+                newimage[i,:,:] = self.transform_1_image(image[i,:,:])
+            return newimage
         if image.ndim == 2 and image.shape != (self.__nx__,self.__ny__):
             newimage = N.reshape(image,(image.shape[0],self.__nx__,self.__ny__))
             for i in range(image.shape[0]):