From d22d0adba61a7f345e77cd195d2472a8859fd27a Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Tr=E1=BA=A7n=20Nam=20Tu=E1=BA=A5n?= Date: Tue, 20 Aug 2024 00:03:46 +0700 Subject: [PATCH] Fix input data for get_scaled_width and get_scaled_height --- py/defs/formatters.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/py/defs/formatters.py b/py/defs/formatters.py index 983d158..d885f89 100644 --- a/py/defs/formatters.py +++ b/py/defs/formatters.py @@ -38,12 +38,12 @@ def convert_skip_clip(stop_at_clip_layer, input_data): def get_scaled_width(scaled_by, input_data): - samples = input_data["samples"][0]["samples"] + samples = input_data[0]["samples"][0]["samples"] return round(samples.shape[3] * scaled_by * 8) def get_scaled_height(scaled_by, input_data): - samples = input_data["samples"][0]["samples"] + samples = input_data[0]["samples"][0]["samples"] return round(samples.shape[2] * scaled_by * 8)