topng return width and height

[myo-class.git] / topng.py

diff --git a/topng.py b/topng.py
index 9651446a4b9f127ee662971a920037556810c68b..c9b72d963fb31594be3d1a350daf771c956c4873 100644 (file)
--- a/topng.py
+++ b/topng.py
@@ -40,6 +40,9 @@ def getdir(filepath):
        return '/'.join(filepath.split('/')[:-1]) + '/'
 
 def topng(inputfile, outfile=None, overwrite=True):
+       """
+               return (64, 64) : the width and height
+       """
        try:
                dicimg = pydicom.read_file(inputfile) # read dicom image
        except pydicom.errors.InvalidDicomError as e:
@@ -48,6 +51,7 @@ def topng(inputfile, outfile=None, overwrite=True):
        img = dicimg.pixel_array# get image array (12bits)
 
        # test <<
+       # return img.shape # $$ COMMENT OR REMOVE THIS LINE 
        print('img', img)
        # print('img', type(img))
        # print('min', img.min(), 'max', img.max())
@@ -82,6 +86,8 @@ def topng(inputfile, outfile=None, overwrite=True):
 
        cv2.imwrite(savepath, img, [cv2.IMWRITE_PNG_COMPRESSION, 0]) # write png image
 
+       # test
+       return img.shape
 
 def topngs(inputdir, outdir):
        """