Spaces:
Runtime error
Runtime error
resize batch
Browse files- kornia_aug.py +1 -0
kornia_aug.py
CHANGED
|
@@ -28,6 +28,7 @@ uploaded_files = st.sidebar.file_uploader("Choose a file",accept_multiple_files=
|
|
| 28 |
if uploaded_files is not None:
|
| 29 |
for uploaded_file in uploaded_files:
|
| 30 |
im = Image.open(uploaded_file)
|
|
|
|
| 31 |
image = F.pil_to_tensor(im).float() / 255
|
| 32 |
ims.append(image)
|
| 33 |
else:
|
|
|
|
| 28 |
if uploaded_files is not None:
|
| 29 |
for uploaded_file in uploaded_files:
|
| 30 |
im = Image.open(uploaded_file)
|
| 31 |
+
im=im.resize((512,512))
|
| 32 |
image = F.pil_to_tensor(im).float() / 255
|
| 33 |
ims.append(image)
|
| 34 |
else:
|