Save model using Keras.

README.md

@@ -1,122 +1,17 @@
+
 ---
 library_name: keras
-license: gpl-3.0
-datasets:
-- logasja/FDF
-pipeline_tag: image-to-image
-widget:
-- text: input
-  output:
-    url: ./assets/input.png
-- text: target
-  output:
-    url: ./assets/target.png
-- text: output
-  output:
-    url: ./assets/output.png
-base_model:
-- vnet
-- logasja/ArcFace
-- logasja/VGGFace2
-metrics:
-- TopIQ-FR
-- ArcFace Cosine Distance
-- VGGFace2 Cosine Distance
-tags:
-- adversarial
-- aesthetic
-- quality
-- filter
 ---
-<Gallery />
-
-Training logs [here](https://wandb.ai/spuds/auramask/runs/95e53b958929dc62e2db048d3f302cfe)
 
-# Model Description
-This model uses a modified vnet for 2D input/output implemented [here](https://github.com/logasja/keras3-unets) with the following configuration.
+This model has been uploaded using the Keras library and can be used with JAX,
+TensorFlow, and PyTorch backends.
 
-```json
-{
-    "activation": "ReLU",
-    "batch_norm": false,
-    "filter_num": [
-        128,
-        256,
-        512,
-        1024,
-        1024
-    ],
-    "n_labels": 3,
-    "output_activation": "tanh",
-    "pool": false,
-    "res_num_ini": 1,
-    "res_num_max": 3,
-    "unpool": false
-}
-```
+This model card has been generated automatically and should be completed by the
+model author.
+See [Model Cards documentation](https://huggingface.co/docs/hub/model-cards) for
+more information.
 
-```json
-{
-    "alpha": 0.0001,
-    "batch": 32,
-    "epochs": 500,
-    "epsilon": 1,
-    "input": "(256, 256)",
-    "losses": {
-        "FEAT_ArcFace": {
-            "d": "cosine_similarity",
-            "f": "ArcFace",
-            "name": "FEAT_ArcFace",
-            "reduction": "sum_over_batch_size",
-            "threshold": 0.68,
-            "weight": 0.05
-        },
-        "FEAT_VGG-Face": {
-            "d": "cosine_similarity",
-            "f": "VGG-Face",
-            "name": "FEAT_VGG-Face",
-            "reduction": "sum_over_batch_size",
-            "threshold": 0.68,
-            "weight": 0.05
-        },
-        "IQASSIMC": {
-            "lower_better": false,
-            "name": "IQASSIMC",
-            "reduction": "sum_over_batch_size",
-            "weight": 0.5
-        },
-        "TopIQ": {
-            "full_ref": true,
-            "lower_better": false,
-            "name": "TopIQ",
-            "reduction": "sum_over_batch_size",
-            "score_range": "~0, ~1",
-            "weight": 0.5
-        }
-    },
-    "mixed_precision": true,
-    "optimizer": {
-        "amsgrad": false,
-        "beta_1": 0.9,
-        "beta_2": 0.999,
-        "clipnorm": null,
-        "clipvalue": null,
-        "ema_momentum": 0.99,
-        "ema_overwrite_frequency": null,
-        "epsilon": 1e-07,
-        "global_clipnorm": null,
-        "gradient_accumulation_steps": null,
-        "learning_rate": 9.999999747378752e-05,
-        "loss_scale_factor": null,
-        "name": "adamw",
-        "use_ema": false,
-        "weight_decay": 0.004
-    },
-    "seed": "BIIIIIGSTRETCH",
-    "testing": 0.01,
-    "training": 0.99
-}
-```
+For more details about the model architecture, check out
+[config.json](./config.json).
 
-## Model Architecture Plot
 ![](./assets/summary_plot.png)

metadata.json

@@ -1 +1 @@
-{"keras_version": "3.8.0", "date_saved": "2025-02-27@18:58:14"}
+{"keras_version": "3.8.0", "date_saved": "2025-03-03@12:50:12"}

model.weights.h5

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:1491e9dbcac73cbbed430659e583e65015c916ffcebf29be5ceb8f3b410a66a5
+oid sha256:0f91ba63aed922ab5e0690b677f7e960d58f806526d7a7b4c78ae52965170840
 size 547716936