def getRawPNGsInDir(path):
fnames = glob.glob(path + "*.png")
pngs = list()
pngnames = list()
for f in fnames:
imi = io.imreadGrey(f)
pngs.append(imi)
pngnames.append(f)
return pngs, pngnames
rout = a4.edgeBasedGreenDemosaic(raw, gos, rosy, rosx, bosy, bosx)
io.imwrite(rout, outputname + "_edgeBasedGreenDemosaic.png")
def testImprovedDemosaic(raw, outputname, gos=1, rosy=1, rosx=1, bosy=0, bosx=0):
rout = a4.improvedDemosaic(raw, gos, rosy, rosx, bosy, bosx)
io.imwrite(rout, outputname + "_improvedDemosaic.png")
def testSergei():
sergeis, sergeiNames = getRawPNGsInDir("data/Sergei/")
scount = 0
for f in sergeis:
io.imwrite(a4.sergeiRGB(f), str("Sergei" + "%03d" % (scount)) + ".png")
scount = scount + 1
testDenoise(getPNGsInDir("data/aligned-ISO400/"), "ISO400-denoise")
testDenoise(getPNGsInDir("data/aligned-ISO3200/"), "ISO3200-denoise")
testSNR(getPNGsInDir("data/aligned-ISO400/"), "ISO400-SNR")
testSNR(getPNGsInDir("data/aligned-ISO3200/"), "ISO3200-SNR")
testAlign()
testAlignAndDenoise(getPNGsInDir("data/green/"), "green-aligned-full")
testBasicGreenz(io.imreadGrey("data/raw/signs-small.png"), "demo")
testBasicDemosaic(io.imreadGrey("data/raw/signs-small.png"), "sign-small", gos=0, rosy=1, rosx=1, bosy=0, bosx=0)
testEdgeBasedGreenDemosaic(
io.imreadGrey("data/raw/signs-small.png"), "sign-small", gos=0, rosy=1, rosx=1, bosy=0, bosx=0
)
testImprovedDemosaic(io.imreadGrey("data/raw/signs-small.png"), "sign-small2", gos=0, rosy=1, rosx=1, bosy=0, bosx=0)
testSergei()
def test_10_basicDemosaick(self):
raw = imageIO.imreadGrey('signs-small.png')
im = basicDemosaick(raw)
imageIO.imwrite(im, 'signs-small-basic.png')
for f in sergeis:
io.imwrite(a4.sergeiRGB(f), str("Sergei" + "%03d" % (scount)) + ".png")
scount = scount + 1
def computeSNR(imList):
return a4.logSNR(imList, 1.0)
# Input data:
iso400 = getPNGsInDir("data/aligned-ISO400/")
iso3200 = getPNGsInDir("data/aligned-ISO3200/")
green = getPNGsInDir("data/green/")
# raw, rawnames = getRawPNGsInDir("data/raw/")
signsm = io.imreadGrey("data/raw/signs-small.png")
testDenoise(iso400, "iso400")
testDenoise(iso3200, "iso3200")
testSNR(iso400, "iso400")
testSNR(iso3200, "iso3200")
testAlign()
testAlignAndDenoise(green, "green")
testBasicDemosaic(signsm, "signSmall")
testEdgeBasedGreenDemosaic(signsm, "signSmall")
testImprovedDemosaic(signsm, "signSmall")
testSergei()
