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:
|