Respect user's pyr_level setting at all resolutions
Previously the user's pyr_level was overridden for >=540p content. Now the setting is always used, with the tooltip recommending values per resolution instead. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
12
inference.py
12
inference.py
@@ -58,22 +58,12 @@ class BiMVFIModel:
|
|||||||
img0 = frame0.to(device)
|
img0 = frame0.to(device)
|
||||||
img1 = frame1.to(device)
|
img1 = frame1.to(device)
|
||||||
|
|
||||||
_, _, h, w = img0.shape
|
|
||||||
if h >= 2160:
|
|
||||||
pyr_level = 7
|
|
||||||
elif h >= 1080:
|
|
||||||
pyr_level = 6
|
|
||||||
elif h >= 540:
|
|
||||||
pyr_level = 5
|
|
||||||
else:
|
|
||||||
pyr_level = self.pyr_level
|
|
||||||
|
|
||||||
time_step_tensor = torch.tensor([time_step], device=device).view(1, 1, 1, 1)
|
time_step_tensor = torch.tensor([time_step], device=device).view(1, 1, 1, 1)
|
||||||
|
|
||||||
result_dict = self.model(
|
result_dict = self.model(
|
||||||
img0=img0, img1=img1,
|
img0=img0, img1=img1,
|
||||||
time_step=time_step_tensor,
|
time_step=time_step_tensor,
|
||||||
pyr_level=pyr_level,
|
pyr_level=self.pyr_level,
|
||||||
)
|
)
|
||||||
|
|
||||||
interp = result_dict["imgt_pred"]
|
interp = result_dict["imgt_pred"]
|
||||||
|
|||||||
2
nodes.py
2
nodes.py
@@ -59,7 +59,7 @@ class LoadBIMVFIModel:
|
|||||||
}),
|
}),
|
||||||
"pyr_level": ("INT", {
|
"pyr_level": ("INT", {
|
||||||
"default": 3, "min": 3, "max": 7, "step": 1,
|
"default": 3, "min": 3, "max": 7, "step": 1,
|
||||||
"tooltip": "Pyramid levels for coarse-to-fine processing. Only used for <540p. Higher res auto-selects: 540p=5, 1080p=6, 4K=7. More levels = captures larger motion but slower.",
|
"tooltip": "Pyramid levels for coarse-to-fine processing. More levels = captures larger motion but slower. Recommended: 3-5 for <540p, 5-6 for 1080p, 6-7 for 4K.",
|
||||||
}),
|
}),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user