Semantic segmentation correctly working!

python/semantic_segmentation.py

@@ -3,7 +3,7 @@
 from python.operation import *
 from scipy import ndimage as ndi
 import matplotlib.pyplot as plt
-
+import matplotlib as mpl
 from skimage.morphology import watershed, disk
 from skimage import data
 from skimage.filters import rank
@@ -22,12 +22,20 @@ def execute(input_pixmap):
         gradient = rank.gradient(denoised, disk(2))
 
         labels = watershed(gradient, markers)
-        plt.subplots_adjust(0, 0, 1, 1)
-        img_plot = plt.imshow(labels, cmap=plt.cm.spectral, interpolation='nearest', alpha=.7, aspect=None)
+
+        figure = plt.figure(frameon=False)
+        ax = plt.Axes(figure, [0., 0., 1., 1.])
+        ax.set_axis_off()
+        figure.add_axes(ax)
+        size = (input_pixmap.width() / 100, input_pixmap.height() / 100)
+        figure.set_size_inches(size)
+        extent = mpl.transforms.Bbox(((0,0), size))
+
+        ax.imshow(labels, cmap=plt.cm.spectral, interpolation='nearest')
 
         buffer = io.BytesIO()
-        img_plot.figure.savefig(buffer, format='png')
+        figure.savefig(buffer,  bbox_inches=extent, pad_inches=0, format='png')
         buffer.seek(0)
         output = Image.open(buffer)
-        # imgplot.show()
-        return QPixmap.fromImage(image_to_qimage(output))
+
+        return QPixmap.fromImage(image_to_qimage(output))