Thomas Chardonnens
commited on
Commit
·
f397840
1
Parent(s):
f55a150
form uint32 to int32
Browse files
app.py
CHANGED
|
@@ -147,17 +147,21 @@ def encrypt(user_id, input_image):
|
|
| 147 |
if input_image is None:
|
| 148 |
raise gr.Error("Please choose an image first.")
|
| 149 |
|
|
|
|
|
|
|
|
|
|
| 150 |
if input_image.shape != (224, 224, 3):
|
| 151 |
input_image_pil = Image.fromarray(input_image)
|
| 152 |
input_image_pil = input_image_pil.resize((224, 224))
|
| 153 |
-
input_image =
|
| 154 |
|
| 155 |
# Convert to grayscale and reshape to (1, 1, 224, 224)
|
| 156 |
-
input_image =
|
| 157 |
input_image = input_image.reshape(1, 1, 224, 224)
|
| 158 |
|
| 159 |
-
# Scale values to 12-bit range (
|
| 160 |
-
input_image = (input_image / 255.0 * 4095).astype(
|
|
|
|
| 161 |
|
| 162 |
# Retrieve the client API
|
| 163 |
client = get_client(user_id)
|
|
|
|
| 147 |
if input_image is None:
|
| 148 |
raise gr.Error("Please choose an image first.")
|
| 149 |
|
| 150 |
+
|
| 151 |
+
import numpy as np
|
| 152 |
+
# Resize image if necessary
|
| 153 |
if input_image.shape != (224, 224, 3):
|
| 154 |
input_image_pil = Image.fromarray(input_image)
|
| 155 |
input_image_pil = input_image_pil.resize((224, 224))
|
| 156 |
+
input_image = np.array(input_image_pil)
|
| 157 |
|
| 158 |
# Convert to grayscale and reshape to (1, 1, 224, 224)
|
| 159 |
+
input_image = np.mean(input_image, axis=2).astype(np.float32)
|
| 160 |
input_image = input_image.reshape(1, 1, 224, 224)
|
| 161 |
|
| 162 |
+
# Scale values to 12-bit range (-2048 to 2047)
|
| 163 |
+
input_image = (input_image / 255.0 * 4095 - 2048).astype(np.int16)
|
| 164 |
+
input_image = np.clip(input_image, -2048, 2047)
|
| 165 |
|
| 166 |
# Retrieve the client API
|
| 167 |
client = get_client(user_id)
|