Hugging Face's logo

BueormLLC
/
AID_small

Image-to-Image
image
autoencoder
Bueorm
Model card Files Community
GFBO commited on
Commit
38cc8cc
·
verified ·
1 Parent(s): 1413845

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -51,7 +51,7 @@ model_path = hf_hub_download(repo_id="BueormLLC/AID_small", filename="model.h5")
51
  autoencoder = tf.keras.models.load_model(model_path, custom_objects={'mse': tf.keras.losses.MeanSquaredError()})
52
 
53
  def add_noise(img, noise_type="gaussian"):
54
- img_np = img.numpy() # Convertir el tensor a un arreglo NumPy
55
  if noise_type == "gaussian":
56
  noisy_img = random_noise(img_np, mode='gaussian', var=0.1)
57
  elif noise_type == "salt_pepper":
 
51
  autoencoder = tf.keras.models.load_model(model_path, custom_objects={'mse': tf.keras.losses.MeanSquaredError()})
52
 
53
  def add_noise(img, noise_type="gaussian"):
54
+ img_np = img.numpy()
55
  if noise_type == "gaussian":
56
  noisy_img = random_noise(img_np, mode='gaussian', var=0.1)
57
  elif noise_type == "salt_pepper":