PengWeixuanSZU commited on
Commit
370e193
·
verified ·
1 Parent(s): 8eeaeaf

debug in app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -188,6 +188,7 @@ def inference_and_return_video(dilation_iterations, num_inference_steps, video_s
188
 
189
  images = np.array(images)
190
  masks = np.array(masks)
 
191
  img_tensor, mask_tensor = preprocess_for_removal(images, masks)
192
  mask_tensor = mask_tensor[:,:,:,:1]
193
 
 
188
 
189
  images = np.array(images)
190
  masks = np.array(masks)
191
+ print(f"line 191 images shape:{images.shape},masks shape:{masks.shape}")
192
  img_tensor, mask_tensor = preprocess_for_removal(images, masks)
193
  mask_tensor = mask_tensor[:,:,:,:1]
194