3 months ago
# Keeps 589 bright, boosts jewelry shine, replaces background, and maps to Casinofi duotone.
import cv2, numpy as np
from google.colab import files
from PIL import Image
# Upload your image when prompted
up = files.upload()
fn = list(up.keys())[0]
img_bgr = cv2.imdecode(np.frombuffer(up[fn], np.uint8), cv2.IMREAD_COLOR)
# --- Palette (BGR) ---
HEX = lambda h: (int(h[5:7],16), int(h[3:5],16), int(h[1:3],16))
SHADOW = np.array(HEX("#0F1011"), np.float32)
MID = np.array(HEX("#8E7A55"), np.float32)
HILITE = np.array(HEX("#E6D2A1"), np.float32)
HILITE_PLUS = np.array(HEX("#EBDDB7"), np.float32) # extra-bright cream for 589
# --- Helper: gradient map (shadow -> mid -> highlight) ---
def gradient_map(gray01):
g = gray01[...,None]
t1 = np.clip(g/0.5, 0, 1)
t2 = np.clip((g-0.5)/0.5, 0, 1)
low = SHADOW*(1-t1) + MID*t1
high = MID*(1-t2) + HILITE*t2
return np.where(g<=0.5, low, high)
hsv = cv2.cvtColor(img_bgr, cv2.COLOR_BGR2HSV)
# --- Masks ---
# Background (yellow) range
bg_mask = cv2.inRange(hsv, (15, 120, 120), (40, 255, 255)) # tune if needed
# Shirt (blue) range – helpful for separate contrast if you want
shirt_mask = cv2.inRange(hsv, (95, 80, 40), (130, 255, 255))
# Numbers “589” (white-ish areas on shirt)
gray = cv2.cvtColor(img_bgr, cv2.COLOR_BGR2GRAY)
num_mask = cv2.threshold(gray, 210, 255, cv2.THRESH_BINARY)[1] # bright white
# Jewelry (gold/yellow highlights)
jew_mask = cv2.inRange(hsv, (12, 60, 120), (30, 255, 255)) # gold tones
# Clean masks a bit
def clean(m, k=3):
kernel = cv2.getStructuringElement(cv2.MORPH_ELLIPSE, (k,k))
m = cv2.morphologyEx(m, cv2.MORPH_OPEN, kernel, iterations=1)
m = cv2.morphologyEx(m, cv2.MORPH_CLOSE, kernel, iterations=1)
return m
bg_mask = clean(bg_mask, 5)
shirt_mask= clean(shirt_mask, 5)
num_mask = clean(num_mask, 3)
jew_mask = clean(jew_mask, 3)
# --- Step 1: Replace background with deep charcoal ---
out = img_bgr.copy()
out[bg_mask>0] = SHADOW
# --- Step 2: Convert subject to Casinofi duotone ---
# Work on non-background regions
subj = out.copy()
subj_mask = (bg_mask==0).astype(np.uint8)*255
subj_gray = cv2.cvtColor(subj, cv2.COLOR_BGR2GRAY).astype(np.float32)/255.0
mapped = gradient_map(subj_gray).astype(np.uint8)
mapped = cv2.bitwise_and(mapped, mapped, mask=subj_mask)
bg_area = cv2.bitwise_and(out, out, mask=bg_mask)
out = cv2.add(mapped, bg_area)
# --- Step 3: Boost numbers “589” to brighter cream and keep edges crisp ---
num_rgb = np.zeros_like(out, dtype=np.uint8)
num_rgb[:] = HILITE_PLUS
num_layer = cv2.bitwise_and(num_rgb, num_rgb, mask=num_mask)
out = cv2.bitwise_and(out, out, mask=cv2.bitwise_not(num_mask))
out = cv2.add(out, num_layer)
# Optional: thin dark stroke around numbers
edges = cv2.Canny(num_mask, 50, 150)
stroke = cv2.dilate(edges, np.ones((2,2), np.uint8), iterations=1)
out[stroke>0] = (out[stroke>0]*0 + SHADOW*0.9).astype(np.uint8)
# --- Step 4: Jewelry shine (Screen-like brighten in cream) ---
# Create a cream layer and blend additively where jewelry mask is
j_layer = np.zeros_like(out, dtype=np.float32)
j_layer[:] = HILITE
j_mask_f = (jew_mask.astype(np.float32)/255.0)[...,None]
out_f = out.astype(np.float32)
out = np.clip(out_f + j_layer*0.35*j_mask_f, 0, 255).astype(np.uint8)
# --- Step 5: Gentle contrast pop on subject only ---
subj_mask3 = cv2.merge([subj_mask, subj_mask, subj_mask])
subj_pix = np.where(subj_mask3>0)
sub = out.astype(np.float32)
sub[subj_pix] = np.clip((sub[subj_pix]-20)*1.08 + 20, 0, 255)
out = sub.astype(np.uint8)
# Save
cv2.imwrite("output_casinofi.png", out)
files.download("output_casinofi.png")
print("Done. Download output_casinofi.png")
