code-loader 0.2.55__py3-none-any.whl → 0.2.56__py3-none-any.whl

code_loader/helpers/detection/yolo/utils.py

@@ -1,5 +1,4 @@
 from typing import Tuple, List, Union
-import collections.abc
 import numpy as np
 import tensorflow as tf  # type: ignore
 from numpy.typing import NDArray
@@ -56,10 +55,10 @@ def scale_loc_prediction(loc_pred: List[tf.Tensor], decoded: bool = False, image
                          strides: Tuple[int, int, int] = (8, 16, 32)) -> \
         List[tf.Tensor]:
     new_loc_pred = [None] * len(loc_pred)
-    if not isinstance(image_size, collections.abc.Sequence):
-        scale_arr: NDArray[np.float32] = np.array([image_size, image_size, image_size, image_size])
+    if isinstance(image_size, int) or isinstance(image_size, float):
+        scale_arr: NDArray[np.float32] = np.array([image_size, image_size, image_size, image_size], dtype=np.float32)
     else:
-        scale_arr = np.array([*image_size[::-1], *image_size[::-1]])         # type: ignore
+        scale_arr = np.array([*image_size[::-1], *image_size[::-1]], dtype=np.float32)
     if decoded:
         new_loc_pred = [loc / scale_arr for loc in loc_pred]
     else:

{code_loader-0.2.55.dist-info → code_loader-0.2.56.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: code-loader
-Version: 0.2.55
+Version: 0.2.56
 Summary: 
 Home-page: https://github.com/tensorleap/code-loader
 License: MIT

{code_loader-0.2.55.dist-info → code_loader-0.2.56.dist-info}/RECORD

@@ -13,7 +13,7 @@ code_loader/helpers/detection/yolo/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQe
 code_loader/helpers/detection/yolo/decoder.py,sha256=EsjebDBya53BXtth3vCwLNAlCVwpaTpG0n8oUB2AJOg,6398
 code_loader/helpers/detection/yolo/grid.py,sha256=xcsHh7P10_PrEavvJm1JAraxteQujqN42RfMP5nYrts,3132
 code_loader/helpers/detection/yolo/loss.py,sha256=DxFv2mkm0w7M1yFKTrYUpnYk0RqDxXiHR1MoFEWtRhk,6959
-code_loader/helpers/detection/yolo/utils.py,sha256=uaKJibwNiSpjCt0SGBDo3A1kiy1y-DueMFy4KVdIIO8,6565
+code_loader/helpers/detection/yolo/utils.py,sha256=eV9-IZhglRNGud1UmSckIuA49WRKJitDQhPKK7iIaBc,6563
 code_loader/leap_binder/__init__.py,sha256=Oe7DNj6RmgX-tMruOrI7GYvf3SyAnzOxugF2e117_Z8,93
 code_loader/leap_binder/leapbinder.py,sha256=TXjLdySIXjujrZrYo3wcF8Q2olYfQ0vschUZ8dHOTLs,6404
 code_loader/leaploader.py,sha256=StIRy3LOB-V0RtC59E72dTN6-DkFyE10xMYpYdvDwzU,14096
@@ -22,7 +22,7 @@ code_loader/metrics/default_metrics.py,sha256=EbRk6LDNgTZXN9Y8dp7wXrbxHR9MVIJcPr
 code_loader/utils.py,sha256=WUcM97OuxrhfLCRPoH9EbXrxajNpYgX1CTMc3_PXtYU,1736
 code_loader/visualizers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 code_loader/visualizers/default_visualizers.py,sha256=HqWx2qfTrroGl2n8Fpmr_4X-rk7tE2oGapjO3gzz4WY,2226
-code_loader-0.2.55.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
-code_loader-0.2.55.dist-info/WHEEL,sha256=vxFmldFsRN_Hx10GDvsdv1wroKq8r5Lzvjp6GZ4OO8c,88
-code_loader-0.2.55.dist-info/METADATA,sha256=ED96CdyskdvjW1e9mcpcUD4IbfUXjoftaeOe0qgg1jo,874
-code_loader-0.2.55.dist-info/RECORD,,
+code_loader-0.2.56.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
+code_loader-0.2.56.dist-info/WHEEL,sha256=vxFmldFsRN_Hx10GDvsdv1wroKq8r5Lzvjp6GZ4OO8c,88
+code_loader-0.2.56.dist-info/METADATA,sha256=H-pEH32AzYUCKPBpTrUUIan0v8urAxyEmlpSNRmMpIo,874
+code_loader-0.2.56.dist-info/RECORD,,