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()