117 lines (117 with data), 12.6 kB

B

6e╔ac>Ń@sB
dd
lm
Z

ddlZddlZddlZdZdZdZyddl	Z	ddl	m
Z

eé
Wnek
rd


dZYnXesödZyddlZWnek
rĺ


dZYnXes┌es┌dZyddlZddlm
Z

Wn ek
rě


dZedâ
é
YnXGdd	äd	eâZGd
dädeâZGdd
äd
eâZGddädeâZddlZddäZdddä
Zdddä
ZdS)Ú)
┌divisionNTF)
┌Image)
┌misczcouldn't load ANY image libraryc@s2eZ
dZd
Zdddä
ZddäZddäZd	d
äZdS)
┌_OtsuPyramidz«segments histogram into pyramid of histograms, each histogram
    half the size of the previous. Also generate omega and mu values
    for each histogram in the pyramid.
    Ú
c	Csát|
â
t
jks&td
tt|
â
â
dâ
é
|
jdkr8tdâ
é
|
|_t
á|
|í\}}ddä|Dâ
}|á|í
\}}}}ddät	|â
Dâ
|_
ddät	|â
Dâ
|_||_d	S)
z$ bins is number of intensity levels z must be passed numpy array. Got z insteadÚz4image must be greyscale (and single value per pixel)c
Ssg|]}
t|
â
ĹqSę)
┌int)┌.0┌
hrr˙6/home/wangqiuli/Code/Newcode1222_monk/features/otsu.py˙
<listcomp>3sz+_OtsuPyramid.load_image.<locals>.<listcomp>c
Ssg|]}
|
ĹqSrr)r
┌omegasrrrr
7sc
Ssg|]}
|
ĹqSrr)r
┌musrrrr
9sN)
┌type┌np┌ndarray┌
ValueError┌str┌ndim┌im┌	histogram┌$_create_histogram_and_stats_pyramids┌reversed┌omegaPyramid┌	muPyramid┌ratioPyramid)	┌selfr┌bins┌histZrangesZhistPyrZomegaPyrZmuPyrZratioPyrrrr┌
load_image$s







z_OtsuPyramid.load_imagec
s└tłâ
}d
ë
t
tá|ł
íâ
}g}g}g}g}xLt|â
D]@}|áłí

çç
fddätd|ł
âDâ
}	|	ë|ł
}|áł
í

q6Wd|d<x0|D](ë|áłí
\}
}}|á|
í

|á|í

qłW||||fS)a│
Expects hist to be a single list of numbers (no numpy array)
        takes an input histogram (with 256 bins) and iteratively
        compresses it by a factor of 2 until the last compressed
        histogram is of size 2. It stores all these generated histograms
        in a list-like pyramid structure. Finally, create corresponding
        omega and mu lists for each histogram and return the 3
        generated pyramids.
        Úc
s g|]}
tł|
|
ł
ůâ
ĹqSr)
┌sum)r
┌
i)r┌ratiorrr
PszE_OtsuPyramid._create_histogram_and_stats_pyramids.<locals>.<listcomp>r
Ú
)┌lenr	┌math┌log┌range┌append┌(_calculate_omegas_and_mus_from_histogram)
rrr┌
reductionsZcompressionFactorZhistPyramidrr┌
_ZreducedHistrr┌muTr)rr$rr<s&	














z1_OtsuPyramid._create_histogram_and_stats_pyramidscCsÉ|á|
í
\}}t
|â
}td
â
}g}x&t|â
D]}|||7}|á|í

q,Wtd
â
}g}	x&t|â
D]}|||7}|	á|í

q`Wt|â
}
||	|
fS)zk Comput histogram statistical data: omega and mu for each
        intensity level in the histogram
        r
)┌ _calculate_histogram_pixel_statsr&┌floatr)r*)rrZprobabilityLevelsZ
meanLevelsrZptotalrr#Zmtotalrr.rrrr+as









z5_OtsuPyramid._calculate_omegas_and_mus_from_histogramcsJtł
â
}t
tł
â
â
ëçç
fd
dät|â
Dâ
ëçf
ddät|â
Dâ
}ł|fS)a
Given a histogram, compute pixel probability and mean
        levels for each bin in the histogram. Pixel probability
        represents the likely-hood that a pixel's intensty resides in
        a specific bin. Pixel mean is the intensity-weighted pixel
        probability.
        c
sg|]}
ł
|
łĹqSrr)r
r#)┌
Nrrrr
çszA_OtsuPyramid._calculate_histogram_pixel_stats.<locals>.<listcomp>c
sg|]}
|
ł|
ĹqSrr)r
r#)
┌hist_probabilityrrr
ës)r&r0r"r))rrrZ
pixel_meanr)r1rr2rr/ys
z-_OtsuPyramid._calculate_histogram_pixel_statsN)
r)┌__name__┌
__module__┌__qualname__┌__doc__r rr+r/rrrrrs

%rc@s2eZ
dZd
ZddäZddäZddäZdd	d
ä
ZdS)┌OtsuFastMultithresholdz╦Sacrifices precision for speed. OtsuFastMultithreshold can dial
    in to the threshold but still has the possibility that its
    thresholds will not be the same as a naive-Otsu's method would give
    c	sÂg|_|á
|
í
}t|j|â
|_|á|
í
}|jd
}xnt|t|jâ
âD]Z}|j|}|j|}t|||â}|á	|í
}|já
|í

|j|ëł}çf
ddä|Dâ
}qFWçf
ddä|Dâ
S)Nr!c
sg|]}
|
łĹqSrr)r
┌
t)
┌scalingrrr
źszAOtsuFastMultithreshold.calculate_k_thresholds.<locals>.<listcomp>c
sg|]}
|
łĹqSrr)r
r8)
r9rrr
»s)Z
threshPyramid┌_get_smallest_fitting_pyramidr&rr┌_get_first_guess_thresholdsr)r┌_ThresholdHunter┌%find_best_thresholds_around_estimatesr*r)	r┌
k┌start┌
thresholds┌deviater#rrZhunterr)
r9r┌calculate_k_thresholdsös















z-OtsuFastMultithreshold.calculate_k_thresholdscCs,x&t|j
â
D]\}}t|â
|
kr|SqWd
S)zXReturn the index for the smallest pyramid set that can fit
        K thresholds
        N)┌	enumeraterr&)rr>r#Zpyramidrrrr:▒s

z4OtsuFastMultithreshold._get_smallest_fitting_pyramidcsf|
d
}|jd
ëçf
ddät
|ddâDâ
łg
çf
ddät
d|âDâ
}|á|jdí

|d|
ůS)	z╗Construct first-guess thresholds based on number of
        thresholds (k) and constraining intensity values. FirstGuesses
        will be centered around middle intensity value.
        r!c
sg|]}
ł|
ĹqSrr)r
r#)
┌midwayrrr
└szFOtsuFastMultithreshold._get_first_guess_thresholds.<locals>.<listcomp>r
Ú    c
sg|]}
ł|
ĹqSrr)r
r#)
rDrrr
┴sr%N)rr)r*)rr>ZkHalfZfirstGuessesr)
rDrr;╣s




z2OtsuFastMultithreshold._get_first_guess_thresholdsNcsń|dkr|j}t
|
â
ëdg
|
dg
}d
g
çf
ddätd
łdâDâ
dg
}tj|tjdŹ}tj|jtjdŹ}xntłdâ
D]^}||}tj|jtj	dŹ}|rĘ||k}||d}	|	r─|||	kM}||}
||
}||7}q~W|S)Nr
c
s g|]}
tdł|
d
â
ĹqS)rr%)
r	)r
r#)
r>rrr
═szDOtsuFastMultithreshold.apply_thresholds_to_image.<locals>.<listcomp>r%Ú )
┌dtype)
rr&r)r┌array┌uint8┌zeros┌shape┌onesZbool8)rr@rZbookendedThresholdsZ
greyValuesZ
finalImager#ZkSmallZbwZkLargeZ	greyLevelZ	greyImager)
r>r┌apply_thresholds_to_imageăs&



(










z0OtsuFastMultithreshold.apply_thresholds_to_image)
N)r3r4r5r6rBr:r;rMrrrrr7Äs
r7c@s2eZ
dZd
Zdddä
ZddäZddäZd	d
äZdS)
r<z^Hunt/deviate around given thresholds in a small region to look
    for a better threshold
    r!cCs t|
|â|_
|j
j|_||_dS)
N)┌_BetweenClassVariance┌sigmaBrrA)rrrrArrr┌__init__Ŕs

z_ThresholdHunter.__init__cCstd
|
ddä|
Dâ
f}|
}d
}xR|á|
d
|j
íD]>}|já|í
}||kr\t|â
t|â
krl|}q.||kr.|}|}q.W|S)zvGiven guesses for best threshold, explore to either side of
        the threshold and return the best result.
        r
c
Ssg|]}
dĹqS)
r
r)r
r8rrrr
˘szJ_ThresholdHunter.find_best_thresholds_around_estimates.<locals>.<listcomp>)┌_jitter_thresholds_generatorrrO┌get_total_variancer")r┌estimatedThresholdsZbestResults┌bestThresholdsZbestVariancer@┌variancerrrr=´s








z6_ThresholdHunter.find_best_thresholds_around_estimatescsPd
dä|
Dâ
}
t|
â
dkr$łá
|
í
Sçf
ddä}tjá||
í}ddä|Dâ
}|S)z~Experimental threshold hunting uses scipy optimize method.
        Finds ok thresholds but doesn't work quite as well
        c
Ssg|]}
t|
â
ĹqSr)
r	)r
r>rrrr


szW_ThresholdHunter.find_best_thresholds_around_estimates_experimental.<locals>.<listcomp>Ú
c

sd
łjá
ddä|Dâ
í
S)NrEc
Ssg|]}
t|
â
ĹqSr)
r	)r
r>rrrr

szi_ThresholdHunter.find_best_thresholds_around_estimates_experimental.<locals>.<lambda>.<locals>.<listcomp>)rOrR)
┌
x)
rrr┌<lambda>
s
zU_ThresholdHunter.find_best_thresholds_around_estimates_experimental.<locals>.<lambda>c
Ssg|]}
t|
â
ĹqSr)
r	)r
r>rrrr

s)r"r=┌scipy┌optimize┌fmin)rrSZfxn_to_minimizerTr)
rr┌2find_best_thresholds_around_estimates_experimental
s



zC_ThresholdHunter.find_best_thresholds_around_estimates_experimentalc
csÍ|
d
}t|
â
dkrVx╝t
|j|jdâD]&}||}||ks*||krHq*|g
V
q*Wn||
ddů}
t|
â
}xft
|j|jdâD]N}||}||ksÇ|||króqÇ|á|
|d|í}x|D]}	|g
|	V
q║WqÇWdS)Nr
r%)r&r)rArQ)
rr@┌min_┌max_Z
pastThresh┌offsetZthresh┌
mZrecursiveGeneratorZotherThresholdsrrrrQ
s"










z-_ThresholdHunter._jitter_thresholds_generatorN)
r!)r3r4r5r6rPr=r\rQrrrrr<Ńs

r<c@s$eZ
dZd
däZddäZddäZdS)rNcCs$|
|_||_
t|â
|_t|â
|_dS)
N)rrr&rr"┌muTotal)rrrrrrrP:
s



z_BetweenClassVariance.__init__cCs`d
g
|
|jdg
}
t
|
â
d}d
}x4t|â
D](}|
|}|
|d}||á||í7}q0W|S)zéFunction will pad the thresholds argument with minimum and
        maximum thresholds to calculate between class variance
        r
r%)rr&r)┌_between_thresholds_variance)rr@Z
numClasses┌sigmar#┌k1┌k2rrrrRA
s




z(_BetweenClassVariance.get_total_variancecCs>|j||j|
}|j
||j
|
}|j}|||d
S)z6to be usedin calculating between-class variances only!r!)rrra)rrdreZomega┌mur.rrrrbP
s


z2_BetweenClassVariance._between_thresholds_varianceN)r3r4r5rPrRrbrrrrrN8
srNc
CsÓd
}
d}xft|j
dâ
D]T}xNt|j
dâ
D]<}||||
krL|||}
||||kr,|||}q,WqWt|
|â
xLt|j
dâ
D]:}x4t|j
dâ
D]"}tá|||í
r×d|||<q×WqŐW|||
|}|dS)NrEi'r
r%gÓo@)r)rK┌printr'┌isnan)┌imgZMaxZMinr#┌
jrrr┌	normalizeY
s














rkr%cCst|á
íâ
S)
zŮ
    ret_num: the number of image return
    img_nums: the number of calculate thresholds
    categories_pixel_nums: the number of pixel categories
    return the images with only categories_pixel_nums types of pixels
    )rk┌copy)riZcategories_pixel_numsrrr┌_otsuk
srmšÓ?šÓ┐cCs tâ}|á
|í

|á||
g|íS)
N)r7r rM)ri┌upperZdown┌
categoriesZotrrr┌otsu_helperâ
s



rr)
r%)rnror%)┌
__future__rr'┌numpyr┌cv2ZUSE_PILZUSE_CV2Z	USE_SCIPYZPILr┌ImportErrorrYr┌RuntimeError┌objectrr7r<rNrkrmrrrrrr┌<module>
sD


























pUU