init commit

.gitattributes

@@ -1,35 +1,36 @@
-*.7z filter=lfs diff=lfs merge=lfs -text
-*.arrow filter=lfs diff=lfs merge=lfs -text
-*.bin filter=lfs diff=lfs merge=lfs -text
-*.bz2 filter=lfs diff=lfs merge=lfs -text
-*.ckpt filter=lfs diff=lfs merge=lfs -text
-*.ftz filter=lfs diff=lfs merge=lfs -text
-*.gz filter=lfs diff=lfs merge=lfs -text
-*.h5 filter=lfs diff=lfs merge=lfs -text
-*.joblib filter=lfs diff=lfs merge=lfs -text
-*.lfs.* filter=lfs diff=lfs merge=lfs -text
-*.mlmodel filter=lfs diff=lfs merge=lfs -text
-*.model filter=lfs diff=lfs merge=lfs -text
-*.msgpack filter=lfs diff=lfs merge=lfs -text
-*.npy filter=lfs diff=lfs merge=lfs -text
-*.npz filter=lfs diff=lfs merge=lfs -text
-*.onnx filter=lfs diff=lfs merge=lfs -text
-*.ot filter=lfs diff=lfs merge=lfs -text
-*.parquet filter=lfs diff=lfs merge=lfs -text
-*.pb filter=lfs diff=lfs merge=lfs -text
-*.pickle filter=lfs diff=lfs merge=lfs -text
-*.pkl filter=lfs diff=lfs merge=lfs -text
-*.pt filter=lfs diff=lfs merge=lfs -text
-*.pth filter=lfs diff=lfs merge=lfs -text
-*.rar filter=lfs diff=lfs merge=lfs -text
-*.safetensors filter=lfs diff=lfs merge=lfs -text
-saved_model/**/* filter=lfs diff=lfs merge=lfs -text
-*.tar.* filter=lfs diff=lfs merge=lfs -text
-*.tar filter=lfs diff=lfs merge=lfs -text
-*.tflite filter=lfs diff=lfs merge=lfs -text
-*.tgz filter=lfs diff=lfs merge=lfs -text
-*.wasm filter=lfs diff=lfs merge=lfs -text
-*.xz filter=lfs diff=lfs merge=lfs -text
-*.zip filter=lfs diff=lfs merge=lfs -text
-*.zst filter=lfs diff=lfs merge=lfs -text
-*tfevents* filter=lfs diff=lfs merge=lfs -text
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ckpt filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.mlmodel filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+*.safetensors filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tar filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zst filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text
+*.keras filter=lfs diff=lfs merge=lfs -text

.gitignore

@@ -0,0 +1 @@
+.venv/

README.md

@@ -1,14 +1,14 @@
----
-title: Portrait Colorizer
-emoji: 📈
-colorFrom: indigo
-colorTo: green
-sdk: gradio
-sdk_version: 5.6.0
-app_file: app.py
-pinned: false
-license: apache-2.0
-short_description: Colorize grayscale portraits.
----
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+---
+title: Portrait Colorizer
+emoji: 📈
+colorFrom: indigo
+colorTo: green
+sdk: gradio
+sdk_version: 5.6.0
+app_file: app.py
+pinned: false
+license: apache-2.0
+short_description: Colorize grayscale portraits.
+---
+
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,199 @@
+import tensorflow as tf
+from keras import models
+import numpy as np
+import gradio as gr
+import cv2
+
+# Load the model
+try:
+    generator = models.load_model("generator.keras")
+    print("Model loaded successfully!")
+except Exception as e:
+    print("Error loading model:", e)
+
+
+# Function to preprocess the image (resize, normalize)
+def preprocess_image(img):
+    img = cv2.resize(img, (256, 256))
+
+    # Convert L to range [-1, 1]
+    img = img.astype("float32")
+    img = (img / 127.5) - 1
+
+    # Convert to tensor
+    img = tf.convert_to_tensor(img, dtype=tf.float32)
+
+    img = tf.expand_dims(img, axis=-1)  # Add image dimension
+    img = tf.expand_dims(img, axis=0)  # Add batch dimension
+
+    return img
+
+
+# Function to postprocess the image (denormalize)
+def postprocess_image(img):
+    img = cv2.cvtColor(((img + 1) * 127.5).numpy().astype(np.uint8), cv2.COLOR_LAB2RGB)
+    return np.uint8(np.clip(img, 0, 255))
+
+
+# Function to adjust brightness
+def adjust_brightness(img, brightness=0.0):
+    # Apply brightness adjustment
+    img = cv2.convertScaleAbs(img, beta=int(brightness * 127.0 / 4.0))
+    return np.uint8(np.clip(img, 0, 255))
+
+
+# Function to adjust contrast
+def adjust_contrast(img, contrast=0.0):
+    # Apply contrast adjustment
+    img = cv2.convertScaleAbs(img, alpha=(contrast * 0.75 + 1.0))
+    return np.uint8(np.clip(img, 0, 255))
+
+
+# Function to adjust hue
+def adjust_hue(img, hue_shift=0.0):
+    # Convert the image to HSV
+    hsv_img = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)
+
+    # Adjust the hue channel (value is between 0 and 179 in OpenCV's HSV)
+    hsv_img[:, :, 0] = (
+        hsv_img[:, :, 0] + hue_shift * 90
+    ) % 180  # Hue is wrapped in OpenCV HSV format
+
+    # Convert back to BGR
+    img = cv2.cvtColor(hsv_img, cv2.COLOR_HSV2BGR)
+
+    return np.uint8(np.clip(img, 0, 255))
+
+
+def adjust_saturation(img, saturation_factor=0.0):
+    # Convert the image to HSV
+    hsv_img = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)
+
+    # Adjust the saturation channel (index 1 in HSV)
+    hsv_img[:, :, 1] = np.clip(hsv_img[:, :, 1] * (saturation_factor + 1.0), 0, 255)
+
+    # Convert back to BGR
+    img = cv2.cvtColor(hsv_img, cv2.COLOR_HSV2BGR)
+
+    return np.uint8(np.clip(img, 0, 255))
+
+
+# Define the inference function
+def colorize_image(input_image):
+    # Preprocess the image for the model
+    preprocessed_image = preprocess_image(input_image)
+
+    # Predict using the model
+    output_ab = generator.predict(preprocessed_image)
+    output = tf.concat([preprocessed_image[0], output_ab[0]], axis=-1)
+
+    # Postprocess the output
+    output_image = postprocess_image(output)
+
+    return output_image
+
+
+# Function to colorize and store the result for further manipulation
+def colorize_and_store(img):
+    # Colorize the image
+    colorized_image = colorize_image(img)
+
+    # Return the colorized image for further manipulation (no model call)
+    return colorized_image, colorized_image
+
+
+def make_grayscale_256(img):
+    img = cv2.resize(img, (256, 256))
+    # img = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
+
+    return img
+
+
+css = """
+h1 {
+    text-align: center;
+    display:block;
+    font-size:5rem;
+}
+p {
+    text-align: center;
+    display:block;
+    font-size:2rem;
+}
+#input-image img {
+    filter: grayscale(1);
+}
+"""
+
+# Gradio Interface
+with gr.Blocks(css=css) as demo:
+    demo.title = "Portrait Colorizer"
+    # Add a title
+    gr.Markdown("# Portrait Colorizer")
+    gr.Markdown(
+        "Upload a grayscale image to colorize it and fine-tune the output using the sliders below."
+    )
+
+    with gr.Row():
+        input_image = gr.Image(
+            type="numpy",
+            label="Grayscale Image",
+            image_mode="L",
+            height=256,
+            width=256,
+            elem_id="input-image",
+        )
+        output_image = gr.Image(
+            type="numpy",
+            label="Colorized Image",
+            image_mode="RGB",
+            height=256,
+            width=256,
+        )
+    process_button = gr.Button("Colorize")
+    bright_slider = gr.Slider(-1.0, 1.0, value=0.0, label="Brightness")
+    cont_slider = gr.Slider(-1.0, 1.0, value=0.0, label="Contrast")
+    sat_slider = gr.Slider(-1.0, 1.0, value=0.0, label="Saturation")
+    hue_slider = gr.Slider(-1.0, 1.0, value=0.0, label="Hue")
+
+    # Initially colorize and display the image when it is uploaded
+    colorized_image = gr.State()
+
+    # Button click triggers processing
+    process_button.click(
+        fn=colorize_and_store,
+        inputs=input_image,
+        outputs=[colorized_image, output_image],
+    )
+
+    # Apply hue adjustment to the stored colorized image (no re-generation)
+    bright_slider.change(
+        fn=adjust_brightness,
+        inputs=[colorized_image, bright_slider],
+        outputs=output_image,  # Update output image
+    )
+
+    # Apply hue adjustment to the stored colorized image (no re-generation)
+    cont_slider.change(
+        fn=adjust_contrast,
+        inputs=[colorized_image, cont_slider],
+        outputs=output_image,  # Update output image
+    )
+
+    # Apply hue adjustment to the stored colorized image (no re-generation)
+    hue_slider.change(
+        fn=adjust_hue,
+        inputs=[colorized_image, hue_slider],
+        outputs=output_image,  # Update output image
+    )
+
+    # Apply saturation adjustment to the stored colorized image (no re-generation)
+    sat_slider.change(
+        fn=adjust_saturation,
+        inputs=[colorized_image, sat_slider],
+        outputs=output_image,  # Update output image
+    )
+
+# Launch the app
+if __name__ == "__main__":
+    demo.launch()

requirements.txt

@@ -0,0 +1,8 @@
+tensorflow[and-cuda]==2.14.0
+keras==2.14.0
+matplotlib==3.7.5
+Pillow==9.0.1
+numpy==1.24.3
+ipython==8.28.0
+opencv-python
+gradio