python シグモイド関数 コントラスト協調

Question

http://kesaranpa.hatenablog.com/entry/2018/02/10/211641
上記のサイトのコードを実行したいです。

import math
import numpy as np
from PIL import Image
import sys
import os

def sigmoid(x):
    a = 5.0
    b = 0.5
    return 1.0 / (1.0 + math.exp(a * (b - x)))

lut = np.empty(256)
sigmoid0 = sigmoid(0.0)
sigmoid1 = sigmoid(1.0)
for i in range(256):
    x = i / 255.0

    x = x ** 2.2

    x = (sigmoid(x) - sigmoid0) / (sigmoid1 - sigmoid0) 

    x = x ** (1.0/2.2) 

    lut[i] = 255.0 * x

for fn in sys.argv[1:]:
    img = np.asarray(Image.open(fn))
    img = lut[img]
    ofn = os.path.splitext(fn)[0] + "-out.png"
    Image.fromarray(np.uint8(img)).save(ofn, quality=95)
    print("%s done." % fn)

とりあえず下記のように変えましたが実行できません。
特にエラーなどは出ていません。

img = np.asarray(Image.open(fn))　⇒　img = np.asarray(Image.open(画像のパス))

他に変えなければいけないとこないでしょうか。

Comments

[y_waiwai]

このままではコードが読めないので、質問を編集し、<code>ボタンを押し、出てくる’’’の枠の中にコードを貼り付けてください

[takashi_1111]

ご指摘ありがとうございます。
枠の中にコードを貼り付けました。

Answer 1

とりあえず変換したい画像パスを直接書いて、実行できることは確認できました。

import math
import numpy as np
from PIL import Image
import sys
import os


def sigmoid(x):
    a = 5.0
    b = 0.5
    return 1.0 / (1.0 + math.exp(a * (b - x)))


lut = np.empty(256)
sigmoid0 = sigmoid(0.0)
sigmoid1 = sigmoid(1.0)
for i in range(256):
    x = i / 255.0
    x = x ** 2.2
    x = (sigmoid(x) - sigmoid0) / (sigmoid1 - sigmoid0)
    x = x ** (1.0 / 2.2)
    lut[i] = 255.0 * x

img_path = "sample.jpg"
img = np.asarray(Image.open(img_path))
img = lut[img]
ofn = os.path.splitext(img_path)[0] + "-out.png"
Image.fromarray(np.uint8(img)).save(ofn, quality=95)

Comments

[takashi_1111]

実行できました。
ありがとうございます。