Update trimap_module.py

trimap_module.py

@@ -133,9 +133,7 @@ def trimap(image, name, size, number, erosion=False):
             if (remake[i,j] != 0 and remake[i,j] != 255):
                 remake[i,j] = 127
 
-    path = "./images/results/"  ## Change the directory
-    new_name = '{}px_'.format(size) + name + '_{}.png'.format(number)
-    cv2.imwrite(os.path.join(path, new_name) , remake)
+    return remake
 
 
 #############################################