depth model

testscases/test.py

@@ -39,9 +39,9 @@ alpha = cv2.imread('sampleinput/alpha.png')
 # mask2 = cv2.imread('sampleinput/mask2.png')
 # out = gimpml.interpolateframe(mask1, mask2, 'output/interpolateframes')
 #
-# face = cv2.imread('sampleinput/face.png')
-# out = gimpml.depth(face[:, :, ::-1])
-# cv2.imwrite('output/tmp-depth.png', out[:, :, ::-1])
+img = cv2.imread('sampleinput/img.png')
+out = gimpml.depth(img[:, :, ::-1])
+cv2.imwrite("output/tmp-depth.png", out.astype("uint16"), [cv2.IMWRITE_PNG_COMPRESSION, 0])
 #
 # image = cv2.imread('sampleinput/face.png')
 # out = gimpml.semseg(image[:, :, ::-1])
@@ -51,7 +51,9 @@ alpha = cv2.imread('sampleinput/alpha.png')
 # out = gimpml.super(image[:, :, ::-1])
 # cv2.imwrite('output/tmp-super.png', out[:, :, ::-1])
 #
-image = cv2.imread('sampleinput/inpaint.png')
-mask = cv2.imread('sampleinput/inpaint-mask.png')
-out = gimpml.inpaint(image[:, :, ::-1], mask[:, :, 0])
-cv2.imwrite('output/tmp-inpaint.png', out[:, :, ::-1])
+# image = cv2.imread('sampleinput/inpaint.png')
+# mask = cv2.imread('sampleinput/inpaint-mask.png')
+# image = cv2.imread(r'C:\Users\Kritik Soman\GIMP-ML\cache1.png')
+# mask = cv2.imread(r'C:\Users\Kritik Soman\GIMP-ML\cache0.png')
+# out = gimpml.inpaint(image[:, :, ::-1], mask[:, :, 0])
+# cv2.imwrite(r'C:\Users\Kritik Soman\GIMP-ML\res.png', out[:, :, ::-1])