Omnibus commited on
Commit
01f4376
1 Parent(s): 5f304a2

Update flood.py

Browse files
Files changed (1) hide show
  1. flood.py +2 -2
flood.py CHANGED
@@ -115,8 +115,8 @@ def the_thing(inputs,input_rand, model_choice,num,top_num,timestr):
115
  output=proc1(inputs)
116
  #output_pro=Image.open(output)
117
  output_pro=cv2.imread(output)
118
- cv2.imwrite(f"./frames{timestr}/frame-{num}.png",output_pro)
119
- frame_file_box.append(Path(f"./frames{timestr}/frame-{num}.png"))
120
  frame_box.append(output)
121
  input_rand=f"{input_rand} "
122
  if int(num) < int(top_num):
 
115
  output=proc1(inputs)
116
  #output_pro=Image.open(output)
117
  output_pro=cv2.imread(output)
118
+ cv2.imwrite(f"./frames{timestr}/frame-{timestr}-{num}.png",output_pro)
119
+ frame_file_box.append(Path(f"./frames{timestr}/frame-{timestr}-{num}.png"))
120
  frame_box.append(output)
121
  input_rand=f"{input_rand} "
122
  if int(num) < int(top_num):