Spaces:
Running
on
Zero
Running
on
Zero
Update FFV1MT_MS.py
Browse files- 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 |
-
|
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)
|