Spaces:
Paused
Paused
Commit
·
82bebc2
1
Parent(s):
e4ea7b7
fix
Browse files- Image3DProcessor.py +1 -3
Image3DProcessor.py
CHANGED
|
@@ -62,10 +62,8 @@ class Image3DProcessor:
|
|
| 62 |
image = set_white_background(image)
|
| 63 |
|
| 64 |
image = resize_to_128(image)
|
| 65 |
-
|
| 66 |
-
return image
|
| 67 |
|
| 68 |
-
|
| 69 |
@torch.no_grad()
|
| 70 |
def reconstruct_and_export(self, image):
|
| 71 |
# Ensure the input image is a NumPy array after preprocessing
|
|
|
|
| 62 |
image = set_white_background(image)
|
| 63 |
|
| 64 |
image = resize_to_128(image)
|
|
|
|
|
|
|
| 65 |
|
| 66 |
+
return image
|
| 67 |
@torch.no_grad()
|
| 68 |
def reconstruct_and_export(self, image):
|
| 69 |
# Ensure the input image is a NumPy array after preprocessing
|