sunana commited on
Commit
4a13402
·
1 Parent(s): 9537131

Update FFV1MT_MS.py

Browse files
Files changed (1) hide show
  1. FFV1MT_MS.py +4 -3
FFV1MT_MS.py CHANGED
@@ -252,11 +252,12 @@ class FFV1DNN(nn.Module):
252
  flows, attn in zip(flows[1:], attn)]
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)
 
252
  flows, attn in zip(flows[1:], attn)]
253
  assert len(flows_bi) == len(flows_up)
254
  results_dict["flow_seq"] = flows_up
255
+
256
+ flows_up = flows_up[:-2]
257
+ attn_viz = attn_viz
258
  print(len(flows_up), len(attn_viz))
259
 
260
+
 
261
  flow = plt_show_img_flow(image_list_ori, flows_up)
262
  h = int(MT_size[0] * y)
263
  w = int(MT_size[1] * x)