keras-nightly 3.12.0.dev2025082703__py3-none-any.whl → 3.12.0.dev2025082803__py3-none-any.whl

keras/src/backend/jax/image.py

@@ -682,7 +682,9 @@ def gaussian_blur(
 ):
     def _create_gaussian_kernel(kernel_size, sigma, dtype):
         def _get_gaussian_kernel1d(size, sigma):
-            x = jnp.arange(size, dtype=dtype) - (size - 1) / 2
+            x = jnp.arange(size, dtype=dtype) - jnp.array(
+                (size - 1) / 2, dtype=dtype
+            )
             kernel1d = jnp.exp(-0.5 * (x / sigma) ** 2)
             return kernel1d / jnp.sum(kernel1d)
 
@@ -697,8 +699,8 @@ def gaussian_blur(
         return kernel
 
     images = convert_to_tensor(images)
-    sigma = convert_to_tensor(sigma)
-    dtype = images.dtype
+    dtype = backend.standardize_dtype(images.dtype)
+    sigma = convert_to_tensor(sigma, dtype=dtype)
 
     if len(images.shape) not in (3, 4):
         raise ValueError(

keras/src/backend/numpy/image.py

@@ -560,6 +560,7 @@ def affine_transform(
             f"{AFFINE_TRANSFORM_FILL_MODES}. Received: fill_mode={fill_mode}"
         )
 
+    images = convert_to_tensor(images)
     transform = convert_to_tensor(transform)
 
     if len(images.shape) not in (3, 4):
@@ -575,10 +576,11 @@ def affine_transform(
             f"transform.shape={transform.shape}"
         )
 
-    # scipy.ndimage.map_coordinates lacks support for half precision.
-    input_dtype = images.dtype
-    if input_dtype == "float16":
-        images = images.astype("float32")
+    # `scipy.ndimage.map_coordinates` lacks support for float16 and bfloat16.
+    input_dtype = backend.standardize_dtype(images.dtype)
+    compute_dtype = backend.result_type(input_dtype, "float32")
+    images = images.astype(compute_dtype)
+    transform = transform.astype(compute_dtype)
 
     # unbatched case
     need_squeeze = False
@@ -622,7 +624,7 @@ def affine_transform(
     # transform the indices
     coordinates = np.einsum("Bhwij, Bjk -> Bhwik", indices, transform)
     coordinates = np.moveaxis(coordinates, source=-1, destination=1)
-    coordinates += np.reshape(offset, newshape=(*offset.shape, 1, 1, 1))
+    coordinates += np.reshape(offset, (*offset.shape, 1, 1, 1))
 
     # apply affine transformation
     affined = np.stack(
@@ -643,9 +645,7 @@ def affine_transform(
         affined = np.transpose(affined, (0, 3, 1, 2))
     if need_squeeze:
         affined = np.squeeze(affined, axis=0)
-    if input_dtype == "float16":
-        affined = affined.astype(input_dtype)
-    return affined
+    return affined.astype(input_dtype)
 
 
 def perspective_transform(
@@ -758,6 +758,14 @@ def perspective_transform(
 
 
 def compute_homography_matrix(start_points, end_points):
+    start_points = convert_to_tensor(start_points)
+    end_points = convert_to_tensor(end_points)
+    dtype = backend.result_type(start_points.dtype, end_points.dtype, float)
+    # `np.linalg.solve` lacks support for float16 and bfloat16.
+    compute_dtype = backend.result_type(dtype, "float32")
+    start_points = start_points.astype(dtype)
+    end_points = end_points.astype(dtype)
+
     start_x1, start_y1 = start_points[:, 0, 0], start_points[:, 0, 1]
     start_x2, start_y2 = start_points[:, 1, 0], start_points[:, 1, 1]
     start_x3, start_y3 = start_points[:, 2, 0], start_points[:, 2, 1]
@@ -892,11 +900,11 @@ def compute_homography_matrix(start_points, end_points):
         axis=-1,
     )
     target_vector = np.expand_dims(target_vector, axis=-1)
-
+    coefficient_matrix = coefficient_matrix.astype(compute_dtype)
+    target_vector = target_vector.astype(compute_dtype)
     homography_matrix = np.linalg.solve(coefficient_matrix, target_vector)
     homography_matrix = np.reshape(homography_matrix, [-1, 8])
-
-    return homography_matrix
+    return homography_matrix.astype(dtype)
 
 
 def map_coordinates(
@@ -950,10 +958,14 @@ def map_coordinates(
         )
     else:
         padded = np.pad(inputs, padding, mode=pad_mode)
+
+    # `scipy.ndimage.map_coordinates` lacks support for float16 and bfloat16.
+    if backend.is_float_dtype(padded.dtype):
+        padded = padded.astype("float32")
     result = scipy.ndimage.map_coordinates(
         padded, shifted_coords, order=order, mode=fill_mode, cval=fill_value
     )
-    return result
+    return result.astype(inputs.dtype)
 
 
 def gaussian_blur(
@@ -979,7 +991,11 @@ def gaussian_blur(
     images = convert_to_tensor(images)
     kernel_size = convert_to_tensor(kernel_size)
     sigma = convert_to_tensor(sigma)
-    input_dtype = images.dtype
+    input_dtype = backend.standardize_dtype(images.dtype)
+    # `scipy.signal.convolve2d` lacks support for float16 and bfloat16.
+    compute_dtype = backend.result_type(input_dtype, "float32")
+    images = images.astype(compute_dtype)
+    sigma = sigma.astype(compute_dtype)
 
     if len(images.shape) not in (3, 4):
         raise ValueError(
@@ -1022,8 +1038,7 @@ def gaussian_blur(
         blurred_images = np.transpose(blurred_images, (0, 3, 1, 2))
     if need_squeeze:
         blurred_images = np.squeeze(blurred_images, axis=0)
-
-    return blurred_images
+    return blurred_images.astype(input_dtype)
 
 
 def elastic_transform(

keras/src/backend/tensorflow/image.py

@@ -761,9 +761,9 @@ def gaussian_blur(
         return kernel
 
     images = convert_to_tensor(images)
-    kernel_size = convert_to_tensor(kernel_size)
-    sigma = convert_to_tensor(sigma)
-    dtype = images.dtype
+    dtype = backend.standardize_dtype(images.dtype)
+    kernel_size = convert_to_tensor(kernel_size, dtype=dtype)
+    sigma = convert_to_tensor(sigma, dtype=dtype)
 
     if len(images.shape) not in (3, 4):
         raise ValueError(

keras/src/backend/torch/image.py

@@ -468,8 +468,9 @@ def perspective_transform(
     data_format = backend.standardize_data_format(data_format)
 
     images = convert_to_tensor(images)
-    start_points = torch.tensor(start_points, dtype=torch.float32)
-    end_points = torch.tensor(end_points, dtype=torch.float32)
+    dtype = backend.standardize_dtype(images.dtype)
+    start_points = convert_to_tensor(start_points, dtype=dtype)
+    end_points = convert_to_tensor(end_points, dtype=dtype)
 
     if interpolation not in AFFINE_TRANSFORM_INTERPOLATIONS.keys():
         raise ValueError(
@@ -525,13 +526,15 @@ def perspective_transform(
         transforms = transforms.repeat(batch_size, 1)
 
     grid_x, grid_y = torch.meshgrid(
-        torch.arange(width, dtype=torch.float32, device=images.device),
-        torch.arange(height, dtype=torch.float32, device=images.device),
+        torch.arange(width, dtype=to_torch_dtype(dtype), device=images.device),
+        torch.arange(height, dtype=to_torch_dtype(dtype), device=images.device),
         indexing="xy",
     )
 
     output = torch.empty(
-        [batch_size, height, width, channels], device=images.device
+        [batch_size, height, width, channels],
+        dtype=to_torch_dtype(dtype),
+        device=images.device,
     )
 
     for i in range(batch_size):
@@ -563,8 +566,13 @@ def perspective_transform(
 
 
 def compute_homography_matrix(start_points, end_points):
-    start_points = convert_to_tensor(start_points, dtype=torch.float32)
-    end_points = convert_to_tensor(end_points, dtype=torch.float32)
+    start_points = convert_to_tensor(start_points)
+    end_points = convert_to_tensor(end_points)
+    dtype = backend.result_type(start_points.dtype, end_points.dtype, float)
+    # `torch.linalg.solve` requires float32.
+    compute_dtype = backend.result_type(dtype, "float32")
+    start_points = cast(start_points, dtype)
+    end_points = cast(end_points, dtype)
 
     start_x1, start_y1 = start_points[:, 0, 0], start_points[:, 0, 1]
     start_x2, start_y2 = start_points[:, 1, 0], start_points[:, 1, 1]
@@ -700,9 +708,11 @@ def compute_homography_matrix(start_points, end_points):
         dim=-1,
     ).unsqueeze(-1)
 
+    coefficient_matrix = cast(coefficient_matrix, compute_dtype)
+    target_vector = cast(target_vector, compute_dtype)
     homography_matrix = torch.linalg.solve(coefficient_matrix, target_vector)
     homography_matrix = homography_matrix.reshape(-1, 8)
-
+    homography_matrix = cast(homography_matrix, dtype)
     return homography_matrix
 
 

keras/src/version.py

@@ -1,7 +1,7 @@
 from keras.src.api_export import keras_export
 
 # Unique source of truth for the version number.
-__version__ = "3.12.0.dev2025082703"
+__version__ = "3.12.0.dev2025082803"
 
 
 @keras_export("keras.version")

{keras_nightly-3.12.0.dev2025082703.dist-info → keras_nightly-3.12.0.dev2025082803.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: keras-nightly
-Version: 3.12.0.dev2025082703
+Version: 3.12.0.dev2025082803
 Summary: Multi-backend Keras
 Author-email: Keras team <keras-users@googlegroups.com>
 License: Apache License 2.0

{keras_nightly-3.12.0.dev2025082703.dist-info → keras_nightly-3.12.0.dev2025082803.dist-info}/RECORD

@@ -126,7 +126,7 @@ keras/regularizers/__init__.py,sha256=542Shphw7W8h4Dyf2rmqMKUECVZ8IVBvN9g1LWhz-b
 keras/saving/__init__.py,sha256=KvL2GZxjvgFgEhvEnkvqjIR9JSNHKz-NWZacXajsjLI,1298
 keras/src/__init__.py,sha256=Gi4S7EiCMkE03PbdGNpFdaUYySWDs_FcAJ8Taz9Y1BE,684
 keras/src/api_export.py,sha256=gXOkBOnmscV013WAc75lc4Up01-Kkg9EylIAT_QWctg,1173
-keras/src/version.py,sha256=_xotlmgpnmC5t-uC86NtDftpNS6M-KZgnTfkEVbAt2w,204
+keras/src/version.py,sha256=pgvqOTZbO0O-oqinoneLoQqTXW3UEODq1wx_FRY_dvI,204
 keras/src/activations/__init__.py,sha256=0nL3IFDB9unlrMz8ninKOWo-uCHasTUpTo1tXZb2u44,4433
 keras/src/activations/activations.py,sha256=mogPggtp4CGldI3VOPNmesRxp6EbiR1_i4KLGaVwzL8,17614
 keras/src/applications/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -164,7 +164,7 @@ keras/src/backend/jax/__init__.py,sha256=l_HMwAZ3oAV4Etnw9RPqbvLYPPs3CZYbgaLd_qy
 keras/src/backend/jax/core.py,sha256=kd2-Q3Nm5zp7YwyO_d6ZFlqUCPKP7_hVFAUrgD_FvIU,21418
 keras/src/backend/jax/distribution_lib.py,sha256=-pm5qtFbzTgg_Z7sxq30x4Mjdopn-6qgXFysWoNLtZo,8663
 keras/src/backend/jax/export.py,sha256=jV2yKQLzYjK72vTJmdNomWPLeNS_lDTCEKzQx_5D_-E,7368
-keras/src/backend/jax/image.py,sha256=9ZibpED_HAY_YOCvPbH-3MBlJEAdlwBz4lJz85wI4-c,30388
+keras/src/backend/jax/image.py,sha256=RiYIalbIaUQdDOGpDZUBk5KNsX94Xqg7iyXGATN9V58,30482
 keras/src/backend/jax/layer.py,sha256=QxZeeiimUulsb3j1h3ncNxIoTYdKPO89s0kP49ZwF-w,194
 keras/src/backend/jax/linalg.py,sha256=dtGHRYCvoVlRX0UwbDDdunA8Vp_mA3sdqoasX4P8SbQ,2532
 keras/src/backend/jax/math.py,sha256=1IEDpdoF8e5ltu3D4wbDQuihzvJHhMXz8W9Z_E-eJqU,9391
@@ -179,7 +179,7 @@ keras/src/backend/jax/trainer.py,sha256=N3Ct5iG2tZi1HjX5sLQYPbxKhqFxaMgHzJpGhWQZ
 keras/src/backend/numpy/__init__.py,sha256=NkIur677fC3eRPT_ZsnF_lgid8JQUEmNXxwjXVEdjw8,1305
 keras/src/backend/numpy/core.py,sha256=ZGTkv0BS03x-qJen5DgPlzIdx4v81sGszFX6jxiPsdU,13515
 keras/src/backend/numpy/export.py,sha256=mXJ8egC2Rl_I-ggYOTe-NbPeeWiv55od39aWZimUheo,351
-keras/src/backend/numpy/image.py,sha256=4GCzP2Z3dXfoB9E-y1gSjWX-Z9Jbh30wupnYiFnNs1g,39407
+keras/src/backend/numpy/image.py,sha256=B3PyAIylUOkeFnflFvfqo56k-3jt1QYKJUc4cveG39o,40463
 keras/src/backend/numpy/layer.py,sha256=dTk7W7ql7vRgll7JbOXK5PlIhQw5VHdpSjKciHd8vec,27
 keras/src/backend/numpy/linalg.py,sha256=H8Bdu8LG6OlzXqx8uVxLmTKKE8s9lMoZHMsM2tW4e04,2417
 keras/src/backend/numpy/math.py,sha256=HdkEA5ro7dtQBTP78GFIgqTFLgNQ49PXHhqI1vLRGfo,10169
@@ -204,7 +204,7 @@ keras/src/backend/tensorflow/__init__.py,sha256=vdmzvlszS9T9PPxnfVJZWAJ2ODU00HIr
 keras/src/backend/tensorflow/core.py,sha256=fDR7T6OCVFsdRzwrY5dAPlGEYRWzo21RH2DXd4nIdGc,22852
 keras/src/backend/tensorflow/distribution_lib.py,sha256=Qm7WgudlRLWo6N_EFocqGUvsa1It6u4eAC7LSjOpKMk,2742
 keras/src/backend/tensorflow/export.py,sha256=TC6ikEIhcqw7IVLPigbOYqspKS6TnTzQoYb-jrNO92c,792
-keras/src/backend/tensorflow/image.py,sha256=vnK2pJ2ywHobOnbDLfVd1BdlmvgbLeR6DVE_pYbdHrc,35571
+keras/src/backend/tensorflow/image.py,sha256=lPFNl9QZW1YH0sqe1UerK119SWhQ4_aDFUgjSA_aLOk,35624
 keras/src/backend/tensorflow/layer.py,sha256=iE6XYSZENEoTpNhoXrEOm7gnIOHwOjETZd_p9J_16f0,4334
 keras/src/backend/tensorflow/linalg.py,sha256=fpzxql1ycXIAks9AvS753aiSoaVqAuM6xbv671BulhQ,8038
 keras/src/backend/tensorflow/math.py,sha256=zTu_7Ff6B2Ro862z_xH0OCmIWbV74DjsO5UnfjYuOUQ,12370
@@ -220,7 +220,7 @@ keras/src/backend/tensorflow/trainer.py,sha256=PXQSWwTtQJWZlntBoi4q_FiYeRf8DOpw1
 keras/src/backend/torch/__init__.py,sha256=0SiJ91WMaE_tO5q1zUsLEnU6hmPTpGKPIOkmIWaHlhk,2131
 keras/src/backend/torch/core.py,sha256=k35_MwjA2AldkzgM48UPewJ5xytL4vKdbLepCsWtc4M,24369
 keras/src/backend/torch/export.py,sha256=yYc5-4JxSiaCkbFWpfCIdcm4dDBv_9uG_uH6JR6oGx0,4909
-keras/src/backend/torch/image.py,sha256=I1RY1MM99gufhxjQArkRXS9ZKf5Ki0uA-HG-D47s7kE,38643
+keras/src/backend/torch/image.py,sha256=eer8LZwDMz3k2Dh5gxeTQfwrxPemM_H6eHMIP3AwRss,39149
 keras/src/backend/torch/layer.py,sha256=htECdpv9ioHWM8_zqQkEdxgDsgLu8XJi5yXgnLl-JFw,2084
 keras/src/backend/torch/linalg.py,sha256=2GUb107BufiHEK2zJ_fkFREo8Y8mo0OqUZLkwNNgOv4,1991
 keras/src/backend/torch/math.py,sha256=g-ElDii2Y_o1-t6BAu2nbS7JH-aPqVS5Fqds8aYzIlg,14324
@@ -597,7 +597,7 @@ keras/utils/bounding_boxes/__init__.py,sha256=jtvQll4u8ZY0Z96HwNhP1nxWEG9FM3gI-6
 keras/utils/legacy/__init__.py,sha256=oSYZz6uS8UxSElRaaJYWJEoweJ4GAasZjnn7fNaOlog,342
 keras/visualization/__init__.py,sha256=UKWmiy6sps4SWlmQi9WX8_Z53cPpLlphz2zIeHdwJpQ,722
 keras/wrappers/__init__.py,sha256=QkS-O5K8qGS7C3sytF8MpmO6PasATpNVGF8qtb7Ojsw,407
-keras_nightly-3.12.0.dev2025082703.dist-info/METADATA,sha256=-atFYY_hEpWu-qQ2tRIYNfvuXPpxwM36-TDG_iLmdR8,5970
-keras_nightly-3.12.0.dev2025082703.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-keras_nightly-3.12.0.dev2025082703.dist-info/top_level.txt,sha256=ptcw_-QuGZ4ZDjMdwi_Z0clZm8QAqFdvzzFnDEOTs9o,6
-keras_nightly-3.12.0.dev2025082703.dist-info/RECORD,,
+keras_nightly-3.12.0.dev2025082803.dist-info/METADATA,sha256=p7tghTVvyvcpKfF2_8SfJVNh1n7uCq1OkN2Ls2P7mfM,5970
+keras_nightly-3.12.0.dev2025082803.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+keras_nightly-3.12.0.dev2025082803.dist-info/top_level.txt,sha256=ptcw_-QuGZ4ZDjMdwi_Z0clZm8QAqFdvzzFnDEOTs9o,6
+keras_nightly-3.12.0.dev2025082803.dist-info/RECORD,,