import cv2
import numpy as np
import os
from pathlib import Path

input_dir = \"photos_brutes\"
output_dir = \"photos_corrigees\"

Path(output_dir).mkdir(exist_ok=True)

# -----------------------------
# Red Channel Compensation
# -----------------------------
def red_channel_compensation(img):

    b,g,r = cv2.split(img)

    b_mean = np.mean(b)
    g_mean = np.mean(g)

    r = r + (g_mean - np.mean(r))*0.6
    r = r + (b_mean - np.mean(r))*0.3

    r = np.clip(r,0,255)

    return cv2.merge([b,g,r.astype(np.uint8)])


# -----------------------------
# Gray World White Balance
# -----------------------------
def gray_world(img):

    img = img.astype(np.float32)

    avg_b = np.mean(img[:,:,0])
    avg_g = np.mean(img[:,:,1])
    avg_r = np.mean(img[:,:,2])

    avg = (avg_b + avg_g + avg_r) \/ 3

    img[:,:,0] *= avg\/avg_b
    img[:,:,1] *= avg\/avg_g
    img[:,:,2] *= avg\/avg_r

    img = np.clip(img,0,255)

    return img.astype(np.uint8)


# -----------------------------
# CLAHE Contrast Enhancement
# -----------------------------
def enhance_contrast(img):

    lab = cv2.cvtColor(img, cv2.COLOR_BGR2LAB)

    l,a,b = cv2.split(lab)

    clahe = cv2.createCLAHE(clipLimit=2.5, tileGridSize=(8,8))
    l = clahe.apply(l)

    lab = cv2.merge((l,a,b))

    return cv2.cvtColor(lab, cv2.COLOR_LAB2BGR)


# -----------------------------
# Pipeline complet
# -----------------------------
def process(img):

    img = red_channel_compensation(img)
    img = gray_world(img)
    img = enhance_contrast(img)

    return img


# -----------------------------
# Batch processing
# -----------------------------
for file in os.listdir(input_dir):

    if file.lower().endswith((\".jpg\",\".jpeg\",\".png\",\".tif\",\".tiff\")):

        path = os.path.join(input_dir,file)

        img = cv2.imread(path)

        if img is None:
            print(\"image ignor\u00e9e:\", file)
            continue

        corrected = process(img)

        outpath = os.path.join(output_dir,file)

        cv2.imwrite(outpath,corrected)

        print(\"processed:\",file)

print(\"Termin\u00e9\")<\/code><\/pre>
<\/div><\/div>

<\/hr>
<\/p>\n\n\n\n<\/span>Instalaci\u00f3n<\/span><\/h3>\n\n\n\nEn una consola OSGeo4W<\/strong>:<\/p>\n\n\n\n
pip install opencv-python numpy<\/pre>\n\n\n\n
\n\n\n\n<\/span>Estructura de carpetas<\/span><\/h3>\n\n\n\nproyecto\/
   script.py
   fotos_brutas\/
   fotos_corregidas\/<\/pre>\n\n\n\n
\n\n\n\n<\/span>Antes \/ Despu\u00e9s<\/span><\/h3>\n\n\n\n<\/a>Antes de la correcci\u00f3n \u2014 Despu\u00e9s de la correcci\u00f3n<\/figcaption><\/figure>\n<\/div>\n\n\n
\n\n\n\n<\/span>Por qu\u00e9 se utiliza este m\u00e9todo en investigaci\u00f3n<\/span><\/h3>\n\n\n\nCorrige tres problemas principales de las im\u00e1genes submarinas:<\/p>\n\n\n\n
Problema<\/th> Correcci\u00f3n<\/th><\/tr><\/thead>
absorci\u00f3n del rojo<\/td> Red Channel Compensation<\/td><\/tr>
dominante azul-verde<\/td> Gray World<\/td><\/tr>
bajo contraste<\/td> CLAHE<\/td><\/tr><\/tbody><\/table><\/figure>\n\n\n\nResultados:<\/p>\n\n\n\n
\ncolores m\u00e1s naturales<\/li>\n\n\n\n
texturas de coral m\u00e1s visibles<\/li>\n\n\n\n
segmentaci\u00f3n autom\u00e1tica m\u00e1s estable<\/li>\n<\/ul>\n\n\n\nEsto es muy \u00fatil para el an\u00e1lisis de arrecifes de coral.<\/p>\n\n\n\n
\n","protected":false},"excerpt":{"rendered":"Bajo el agua, las diferentes longitudes de onda de la luz se absorben progresivamente: el rojo desaparece primero, seguido por el naranja y el amarillo. Incluso a poca profundidad, las fotos suelen presentar una dominante azul-cian.…<\/p>\n","protected":false},"author":1,"featured_media":0,"comment_status":"open","ping_status":"closed","sticky":false,"template":"","format":"standard","meta":{"give_campaign_id":0,"_bbp_topic_count":0,"_bbp_reply_count":0,"_bbp_total_topic_count":0,"_bbp_total_reply_count":0,"_bbp_voice_count":0,"_bbp_anonymous_reply_count":0,"_bbp_topic_count_hidden":0,"_bbp_reply_count_hidden":0,"_bbp_forum_subforum_count":0,"sfsi_plus_gutenberg_text_before_share":"","sfsi_plus_gutenberg_show_text_before_share":"","sfsi_plus_gutenberg_icon_type":"","sfsi_plus_gutenberg_icon_alignemt":"","sfsi_plus_gutenburg_max_per_row":"","_monsterinsights_skip_tracking":false,"_monsterinsights_sitenote_active":false,"_monsterinsights_sitenote_note":"","_monsterinsights_sitenote_category":0,"_jetpack_newsletter_access":"","_jetpack_dont_email_post_to_subs":false,"_jetpack_newsletter_tier_id":0,"_jetpack_memberships_contains_paywalled_content":false,"_jetpack_memberships_contains_paid_content":false,"footnotes":"","jetpack_post_was_ever_published":false},"categories":[3821],"tags":[4472,4474,4476,4478,4480,2797,4482],"class_list":["post-16492","post","type-post","status-publish","format-standard","hentry","category-images-es","tag-computer-vision-en-es","tag-coral-reefs-es","tag-image-processing-en-es","tag-marine-science-en-es","tag-opencv-en-es","tag-python-es","tag-underwater-photography-en-es"],"aioseo_notices":[],"jetpack_featured_media_url":"","jetpack_shortlink":"https:\/\/wp.me\/p6XU0A-4i0","jetpack_sharing_enabled":true,"_links":{"self":[{"href":"https:\/\/www.sigterritoires.fr\/index.php\/wp-json\/wp\/v2\/posts\/16492","targetHints":{"allow":["GET"]}}],"collection":[{"href":"https:\/\/www.sigterritoires.fr\/index.php\/wp-json\/wp\/v2\/posts"}],"about":[{"href":"https:\/\/www.sigterritoires.fr\/index.php\/wp-json\/wp\/v2\/types\/post"}],"author":[{"embeddable":true,"href":"https:\/\/www.sigterritoires.fr\/index.php\/wp-json\/wp\/v2\/users\/1"}],"replies":[{"embeddable":true,"href":"https:\/\/www.sigterritoires.fr\/index.php\/wp-json\/wp\/v2\/comments?post=16492"}],"version-history":[{"count":0,"href":"https:\/\/www.sigterritoires.fr\/index.php\/wp-json\/wp\/v2\/posts\/16492\/revisions"}],"wp:attachment":[{"href":"https:\/\/www.sigterritoires.fr\/index.php\/wp-json\/wp\/v2\/media?parent=16492"}],"wp:term":[{"taxonomy":"category","embeddable":true,"href":"https:\/\/www.sigterritoires.fr\/index.php\/wp-json\/wp\/v2\/categories?post=16492"},{"taxonomy":"post_tag","embeddable":true,"href":"https:\/\/www.sigterritoires.fr\/index.php\/wp-json\/wp\/v2\/tags?post=16492"}],"curies":[{"name":"wp","href":"https:\/\/api.w.org\/{rel}","templated":true}]}}

Problema<\/th>	Correcci\u00f3n<\/th><\/tr><\/thead>
absorci\u00f3n del rojo<\/td>	Red Channel Compensation<\/td><\/tr>
dominante azul-verde<\/td>	Gray World<\/td><\/tr>
bajo contraste<\/td>	CLAHE<\/td><\/tr><\/tbody><\/table><\/figure>\n\n\n\n Resultados:<\/p>\n\n\n\n \n colores m\u00e1s naturales<\/li>\n\n\n\n texturas de coral m\u00e1s visibles<\/li>\n\n\n\n segmentaci\u00f3n autom\u00e1tica m\u00e1s estable<\/li>\n<\/ul>\n\n\n\n Esto es muy \u00fatil para el an\u00e1lisis de arrecifes de coral.<\/p>\n\n\n\n \n","protected":false},"excerpt":{"rendered":" Bajo el agua, las diferentes longitudes de onda de la luz se absorben progresivamente: el rojo desaparece primero, seguido por el naranja y el amarillo. Incluso a poca profundidad, las fotos suelen presentar una dominante azul-cian.…<\/p>\n","protected":false},"author":1,"featured_media":0,"comment_status":"open","ping_status":"closed","sticky":false,"template":"","format":"standard","meta":{"give_campaign_id":0,"_bbp_topic_count":0,"_bbp_reply_count":0,"_bbp_total_topic_count":0,"_bbp_total_reply_count":0,"_bbp_voice_count":0,"_bbp_anonymous_reply_count":0,"_bbp_topic_count_hidden":0,"_bbp_reply_count_hidden":0,"_bbp_forum_subforum_count":0,"sfsi_plus_gutenberg_text_before_share":"","sfsi_plus_gutenberg_show_text_before_share":"","sfsi_plus_gutenberg_icon_type":"","sfsi_plus_gutenberg_icon_alignemt":"","sfsi_plus_gutenburg_max_per_row":"","_monsterinsights_skip_tracking":false,"_monsterinsights_sitenote_active":false,"_monsterinsights_sitenote_note":"","_monsterinsights_sitenote_category":0,"_jetpack_newsletter_access":"","_jetpack_dont_email_post_to_subs":false,"_jetpack_newsletter_tier_id":0,"_jetpack_memberships_contains_paywalled_content":false,"_jetpack_memberships_contains_paid_content":false,"footnotes":"","jetpack_post_was_ever_published":false},"categories":[3821],"tags":[4472,4474,4476,4478,4480,2797,4482],"class_list":["post-16492","post","type-post","status-publish","format-standard","hentry","category-images-es","tag-computer-vision-en-es","tag-coral-reefs-es","tag-image-processing-en-es","tag-marine-science-en-es","tag-opencv-en-es","tag-python-es","tag-underwater-photography-en-es"],"aioseo_notices":[],"jetpack_featured_media_url":"","jetpack_shortlink":"https:\/\/wp.me\/p6XU0A-4i0","jetpack_sharing_enabled":true,"_links":{"self":[{"href":"https:\/\/www.sigterritoires.fr\/index.php\/wp-json\/wp\/v2\/posts\/16492","targetHints":{"allow":["GET"]}}],"collection":[{"href":"https:\/\/www.sigterritoires.fr\/index.php\/wp-json\/wp\/v2\/posts"}],"about":[{"href":"https:\/\/www.sigterritoires.fr\/index.php\/wp-json\/wp\/v2\/types\/post"}],"author":[{"embeddable":true,"href":"https:\/\/www.sigterritoires.fr\/index.php\/wp-json\/wp\/v2\/users\/1"}],"replies":[{"embeddable":true,"href":"https:\/\/www.sigterritoires.fr\/index.php\/wp-json\/wp\/v2\/comments?post=16492"}],"version-history":[{"count":0,"href":"https:\/\/www.sigterritoires.fr\/index.php\/wp-json\/wp\/v2\/posts\/16492\/revisions"}],"wp:attachment":[{"href":"https:\/\/www.sigterritoires.fr\/index.php\/wp-json\/wp\/v2\/media?parent=16492"}],"wp:term":[{"taxonomy":"category","embeddable":true,"href":"https:\/\/www.sigterritoires.fr\/index.php\/wp-json\/wp\/v2\/categories?post=16492"},{"taxonomy":"post_tag","embeddable":true,"href":"https:\/\/www.sigterritoires.fr\/index.php\/wp-json\/wp\/v2\/tags?post=16492"}],"curies":[{"name":"wp","href":"https:\/\/api.w.org\/{rel}","templated":true}]}}

<\/span>Script Python<\/span><\/h2>\n\n\n\n<\/div>