Spaces:
Running
Running
Bachmann Roman Christian
commited on
Commit
•
2bde7d7
1
Parent(s):
81c008f
bicubic interpolation
Browse files
app.py
CHANGED
|
@@ -297,7 +297,7 @@ def inference(img, num_tokens, manual_mode, num_rgb, num_depth, num_semseg, seed
|
|
| 297 |
# Center crop and resize RGB
|
| 298 |
image_size = 224 # Train resolution
|
| 299 |
img = TF.center_crop(TF.to_tensor(im), min(im.size))
|
| 300 |
-
img = TF.resize(img, image_size)
|
| 301 |
|
| 302 |
# Predict depth and semseg
|
| 303 |
depth = predict_depth(img)
|
|
|
|
| 297 |
# Center crop and resize RGB
|
| 298 |
image_size = 224 # Train resolution
|
| 299 |
img = TF.center_crop(TF.to_tensor(im), min(im.size))
|
| 300 |
+
img = TF.resize(img, image_size, interpolation=TF.InterpolationMode.BICUBIC)
|
| 301 |
|
| 302 |
# Predict depth and semseg
|
| 303 |
depth = predict_depth(img)
|