sunana commited on
Commit
9537131
1 Parent(s): afb633b

Update FFV1MT_MS.py

Browse files
Files changed (1) hide show
  1. FFV1MT_MS.py +4 -2
FFV1MT_MS.py CHANGED
@@ -253,8 +253,10 @@ class FFV1DNN(nn.Module):
253
  assert len(flows_bi) == len(flows_up)
254
  results_dict["flow_seq"] = flows_up
255
 
256
- flows_up = [flows_up]
257
- attn_viz = [attn_viz]
 
 
258
  flow = plt_show_img_flow(image_list_ori, flows_up)
259
  h = int(MT_size[0] * y)
260
  w = int(MT_size[1] * x)
 
253
  assert len(flows_bi) == len(flows_up)
254
  results_dict["flow_seq"] = flows_up
255
 
256
+ print(len(flows_up), len(attn_viz))
257
+
258
+ flows_up = flows_up
259
+ attn_viz = attn_viz
260
  flow = plt_show_img_flow(image_list_ori, flows_up)
261
  h = int(MT_size[0] * y)
262
  w = int(MT_size[1] * x)