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