Upload folder using huggingface_hub

README.md

@@ -0,0 +1,19 @@
+---
+license: mit
+tags:
+  - cyclegan
+  - image-to-image
+  - gan
+  - pytorch
+---
+
+# CycleGAN apple2orange
+
+CycleGAN model for unpaired image-to-image translation between apples and oranges.
+
+## Files
+- `checkpoint.pt` — trained weights
+- `model_config.json` — architecture parameters
+
+## Usage
+Custom PyTorch implementation. Inference code will be added later.

checkpoint.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:eb247ae7a138b9aeb1fe7bc9656ba9a7b57ca356f743cef6faf597c48cf5ee06
+size 379914573

model_config.json

@@ -0,0 +1,48 @@
+{
+  "model_type": "CycleGAN",
+  "checkpoint_file": "checkpoint.pt",
+  "checkpoint_keys": {
+    "model_state_dict": "model_state_dict"
+  },
+  "domains": {
+    "A": "apple",
+    "B": "orange"
+  },
+  "directions": {
+    "A_to_B": "apple_to_orange",
+    "B_to_A": "orange_to_apple"
+  },
+  "image": {
+    "image_size": 256,
+    "channels": 3
+  },
+  "preprocessing": {
+    "resize": null,
+    "center_crop": null,
+    "to_tensor": true
+  },
+  "normalization": {
+    "mean_A": [0.5921, 0.4201, 0.3659],
+    "std_A": [0.2339, 0.2732, 0.2590],
+    "mean_B": [0.6526, 0.4834, 0.2991],
+    "std_B": [0.2155, 0.2118, 0.2252]
+  },
+  "postprocessing": {
+    "denormalize_A": {
+      "mean": [0.5921, 0.4201, 0.3659],
+      "std": [0.2339, 0.2732, 0.2590]
+    },
+    "denormalize_B": {
+      "mean": [0.6526, 0.4834, 0.2991],
+      "std": [0.2155, 0.2118, 0.2252]
+    }
+  },
+  "model_params": {
+    "gen_d_base": 64,
+    "gen_n_down": 2,
+    "gen_n_blocks": 9,
+    "dis_d_in": 3,
+    "dis_d_base": 64,
+    "dis_n_down": 3
+  }
+}