Update app.py
Browse files
app.py
CHANGED
|
@@ -25,13 +25,13 @@ def make_anaglyph(left_img, right_img, color_method):
|
|
| 25 |
right = right.resize(left.size, Image.LANCZOS)
|
| 26 |
|
| 27 |
# Create a copy of the left image to modify
|
| 28 |
-
result = left.copy()
|
| 29 |
|
| 30 |
# Get the pixel maps
|
| 31 |
width, height = left.size
|
| 32 |
leftMap = left.load()
|
| 33 |
rightMap = right.load()
|
| 34 |
-
resultMap = result.load()
|
| 35 |
|
| 36 |
# Use the selected color matrix
|
| 37 |
m = matrices[color_method]
|
|
@@ -41,14 +41,14 @@ def make_anaglyph(left_img, right_img, color_method):
|
|
| 41 |
for x in range(0, width):
|
| 42 |
r1, g1, b1 = leftMap[x, y]
|
| 43 |
r2, g2, b2 = rightMap[x, y]
|
| 44 |
-
|
| 45 |
int(r1*m[0][0] + g1*m[0][1] + b1*m[0][2] + r2*m[1][0] + g2*m[1][1] + b2*m[1][2]),
|
| 46 |
int(r1*m[0][3] + g1*m[0][4] + b1*m[0][5] + r2*m[1][3] + g2*m[1][4] + b2*m[1][5]),
|
| 47 |
int(r1*m[0][6] + g1*m[0][7] + b1*m[0][8] + r2*m[1][6] + g2*m[1][7] + b2*m[1][8])
|
| 48 |
)
|
| 49 |
|
| 50 |
# Convert back to numpy array for Gradio
|
| 51 |
-
return np.array(
|
| 52 |
|
| 53 |
def make_stereopair(left_img, right_img, color_method):
|
| 54 |
"""Generate a stereo pair from left and right images"""
|
|
|
|
| 25 |
right = right.resize(left.size, Image.LANCZOS)
|
| 26 |
|
| 27 |
# Create a copy of the left image to modify
|
| 28 |
+
#result = left.copy()
|
| 29 |
|
| 30 |
# Get the pixel maps
|
| 31 |
width, height = left.size
|
| 32 |
leftMap = left.load()
|
| 33 |
rightMap = right.load()
|
| 34 |
+
#resultMap = result.load()
|
| 35 |
|
| 36 |
# Use the selected color matrix
|
| 37 |
m = matrices[color_method]
|
|
|
|
| 41 |
for x in range(0, width):
|
| 42 |
r1, g1, b1 = leftMap[x, y]
|
| 43 |
r2, g2, b2 = rightMap[x, y]
|
| 44 |
+
leftMap[x, y] = (
|
| 45 |
int(r1*m[0][0] + g1*m[0][1] + b1*m[0][2] + r2*m[1][0] + g2*m[1][1] + b2*m[1][2]),
|
| 46 |
int(r1*m[0][3] + g1*m[0][4] + b1*m[0][5] + r2*m[1][3] + g2*m[1][4] + b2*m[1][5]),
|
| 47 |
int(r1*m[0][6] + g1*m[0][7] + b1*m[0][8] + r2*m[1][6] + g2*m[1][7] + b2*m[1][8])
|
| 48 |
)
|
| 49 |
|
| 50 |
# Convert back to numpy array for Gradio
|
| 51 |
+
return np.array(left)
|
| 52 |
|
| 53 |
def make_stereopair(left_img, right_img, color_method):
|
| 54 |
"""Generate a stereo pair from left and right images"""
|