ceyda commited on
Commit
002da93
Β·
1 Parent(s): bae54db

resize batch

Browse files
Files changed (1) hide show
  1. 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: