diff --git a/evaluation/eval_lr.py b/evaluation/eval_lr.py index 6531c0a..a11b68a 100644 --- a/evaluation/eval_lr.py +++ b/evaluation/eval_lr.py @@ -98,7 +98,7 @@ class Evaluator: # Write the header if row == 0: metricsheet.write(1, 0, 'Average') - metricsheet.write(1, 1, f'=AVERAGE(C2:ZZ2)') + metricsheet.write(1, 1, '=AVERAGE(C2:ZZ2)') for col in range(len(metric)): metricsheet.write(0, col + 2, col) colname = xlsxwriter.utility.xl_col_to_name(col + 2) diff --git a/inference_matanyone2.py b/inference_matanyone2.py index 66cc773..e9fbcff 100644 --- a/inference_matanyone2.py +++ b/inference_matanyone2.py @@ -8,7 +8,7 @@ from PIL import Image import torch import torch.nn.functional as F -from hugging_face.tools.download_util import load_file_from_url +from matanyone2.utils.download_util import load_file_from_url from matanyone2.utils.inference_utils import gen_dilate, gen_erosion, read_frame_from_videos from matanyone2.inference.inference_core import InferenceCore diff --git a/matanyone2/utils/device.py b/matanyone2/utils/device.py index 0bad062..5a0f573 100644 --- a/matanyone2/utils/device.py +++ b/matanyone2/utils/device.py @@ -1,3 +1,4 @@ +import contextlib import torch import functools @@ -22,7 +23,6 @@ def safe_autocast_decorator(enabled=True): return wrapper return decorator -import contextlib @contextlib.contextmanager def safe_autocast(enabled=True): device = get_default_device()