Update app.py
Browse files
app.py
CHANGED
@@ -78,16 +78,16 @@ def warpImage(im, vx, vy, cast_uint8=True):
|
|
78 |
xx, yy = np.meshgrid(x, y)
|
79 |
XX, YY = np.meshgrid(X, Y)
|
80 |
#XX = XX + vx
|
81 |
-
XX = np.concatenate([XX, vx])
|
82 |
#YY = YY + vy
|
83 |
-
YY = np.concatenate([YY, vy])
|
84 |
mask = (XX < 1) | (XX > width2) | (YY < 1) | (YY > height2)
|
85 |
XX = np.clip(XX, 1, width2)
|
86 |
YY = np.clip(XX, 1, height2)
|
87 |
|
88 |
warpI2 = np.zeros((height1, width1, nChannels))
|
89 |
for i in range(nChannels):
|
90 |
-
f = interp2d(x, y, im[:, :, i], '
|
91 |
foo = f(X, Y)
|
92 |
foo[mask] = 0.6
|
93 |
warpI2[:, :, i] = foo
|
|
|
78 |
xx, yy = np.meshgrid(x, y)
|
79 |
XX, YY = np.meshgrid(X, Y)
|
80 |
#XX = XX + vx
|
81 |
+
XX = np.concatenate([XX, vx], axis = 0)
|
82 |
#YY = YY + vy
|
83 |
+
YY = np.concatenate([YY, vy], axis = 0)
|
84 |
mask = (XX < 1) | (XX > width2) | (YY < 1) | (YY > height2)
|
85 |
XX = np.clip(XX, 1, width2)
|
86 |
YY = np.clip(XX, 1, height2)
|
87 |
|
88 |
warpI2 = np.zeros((height1, width1, nChannels))
|
89 |
for i in range(nChannels):
|
90 |
+
f = interp2d(x, y, im[:, :, i], 'cubic')
|
91 |
foo = f(X, Y)
|
92 |
foo[mask] = 0.6
|
93 |
warpI2[:, :, i] = foo
|