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)
|