Update celle/utils.py
Browse files- celle/utils.py +1 -2
celle/utils.py
CHANGED
|
@@ -198,8 +198,7 @@ def replace_outliers(image, percentile=0.0001):
|
|
| 198 |
return image
|
| 199 |
|
| 200 |
|
| 201 |
-
def process_image(image, dataset, image_type=None):
|
| 202 |
-
image = TF.to_tensor(image)[0].unsqueeze(0).unsqueeze(0)
|
| 203 |
image /= image.max()
|
| 204 |
|
| 205 |
if dataset == "HPA":
|
|
|
|
| 198 |
return image
|
| 199 |
|
| 200 |
|
| 201 |
+
def process_image(image, dataset=None, image_type=None):
|
|
|
|
| 202 |
image /= image.max()
|
| 203 |
|
| 204 |
if dataset == "HPA":
|