From 4ffaef8891627391285de6b4e1ba3e4e94631f66 Mon Sep 17 00:00:00 2001 From: kaanaksit Date: Tue, 19 Dec 2023 21:07:03 +0000 Subject: [PATCH] Cosmetic changes. --- odak/learn/tools/matrix.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/odak/learn/tools/matrix.py b/odak/learn/tools/matrix.py index 237cc6ac..313288d4 100644 --- a/odak/learn/tools/matrix.py +++ b/odak/learn/tools/matrix.py @@ -3,7 +3,7 @@ import torch.nn -def quantize(image_field, bits = 4, limits = [0., 1.]): +def quantize(image_field, bits = 8, limits = [0., 1.]): """ Definition to quantize a image field (0-255, 8 bit) to a certain bits level. @@ -12,7 +12,7 @@ def quantize(image_field, bits = 4, limits = [0., 1.]): image_field : torch.tensor Input image field between any range. bits : int - A value in between 0 to 8. Can not be zero. + A value in between one to eight. limits : list The minimum and maximum of the image_field variable.