X-Git-Url: https://bilbo.iut-bm.univ-fcomte.fr/and/gitweb/myo-class.git/blobdiff_plain/bdfe02afe03428e59b3c85b6cae19d579d82f55c..eeb9b42c3779bbb098c07e2dd91d40ecf6e63372:/topng.py

diff --git a/topng.py b/topng.py
index fde569b..672fd6d 100644
--- a/topng.py
+++ b/topng.py
@@ -27,6 +27,7 @@ INPUT_DIR = '../../Data/Images_anonymous/Case_0002/'
 OUT_DIR = './generated/'
 
 CROP_SIZE = (45, 45) # (width, height)
+EPI_MIN_PIXELS = 30 * 30 # nombre pixels minimal pour considérer un epicardic, else on ignore, 30 * 30, vu que c'est carré
 RED_COLOR = 100
 
 def roundall(*l):
@@ -108,6 +109,43 @@ def getxy(file):
 		return np.array(tmp, dtype=np.int32)
 		# return tmp
 
+def getimgname(file):
+	"""
+		{
+			'Image00001': [{
+				'color': '#ff0000',
+				'points': [
+					{'x': 94.377, 'y': 137.39},
+					{'x': 100.38, 'y': 139.55},
+					{'x': 103.26, 'y': 142.67},
+					{'x': 105.91, 'y': 147.95},
+					{'x': 105.42, 'y': 152.76},
+					{'x': 100.62, 'y': 156.84},
+					{'x': 95.338, 'y': 159.96},
+					{'x': 89.573, 'y': 158.52},
+					{'x': 84.53, 'y': 153},
+					{'x': 82.848, 'y': 149.15},
+					{'x': 82.368, 'y': 142.91},
+					{'x': 85.01, 'y': 138.11},
+					{'x': 89.813, 'y': 137.39},
+					{'x': 94.377, 'y': 137.39}
+				]
+			}]
+		}
+		return 'Image00001' or return ''
+	"""
+	imgXXX = None
+	# print("file", file)
+	with open(file) as jsonfile:
+		data = json.load(jsonfile)
+		if len(data) > 1:
+			print("more than one image in same json")
+		# print("data")
+		# pprint(data)
+		for imgXXX in data: pass  # get the value of key ~= "Image00001", cause it's dynamic
+	return imgXXX
+
+
 def minmax(file):
 	r = getxy(file)
 	if r is not None:
@@ -285,7 +323,7 @@ def readpng(inputfile):# just a specific func to preview a "shape = (X,Y,3)" ima
 		for i, row in enumerate(tab):
 			print(row[0]*65536 + row[0]*256 + row[0], end=" " if i % image.shape[0] != 0 else "\n")
 
-def topng(inputfile, outfile=None, overwrite=True, verbose=False, epimask='', endomask='', centercrop=None, blackmask=False, square=False, redcontour=False):
+def topng(inputfile, outfile=None, overwrite=True, verbose=False, epimask='', endomask='', centercrop=None, blackmask=False, square=False, redcontour=False, epiminpixels=-1):
 	"""
 		(verbose) return (64, 64) : the width and height
 		(not verbose) return (img, dicimg) : the image and the dicimg objects
@@ -348,9 +386,18 @@ def topng(inputfile, outfile=None, overwrite=True, verbose=False, epimask='', en
 				if endomask:
 					img = hollowmask(img, epimask, endomask)
 				else:
+					# ignore small epicardic --<<
+					xmin, ymin, xmax, ymax = minmax(epimask)
+					pixels = (xmax - xmin) * (ymax - ymin)
+					if pixels <= epiminpixels:
+						print( "small epicardic ({}), ignored!".format(inputfile.split('/')[-1]) )
+						return
+					# ignore small epicardic -->>
+
 					img = mask(img, epimask)
 
 		if centercrop:
+
 			img = crop(img, epimask, centercrop)
 
 
@@ -376,6 +423,7 @@ def topng(inputfile, outfile=None, overwrite=True, verbose=False, epimask='', en
 	# np.savetxt(savepath + '.npy', img)
 	# test >>
 
+
 	if np.count_nonzero(img) > 0: # matrix not full of zero
 		cv2.imwrite(savepath, img, [cv2.IMWRITE_PNG_COMPRESSION, 0]) # write png image