def removeSpurs1(bw, minBranchSize):
pruned = morph.mmthin(bw, morph.mmendpoints("homotopic"),minBranchSize)
trimmings = bw - pruned
big_trimmings = morph.mmareaopen(trimmings,minBranchSize)
small_trimmings = trimmings - big_trimmings
#import cv2
#cv2.imshow('hi',small_trimmings.astype('uint8')*100)
#cv2.waitKey(10)
return bw - small_trimmings
import os
os.system(
"wget http://www.eecs.berkeley.edu/Research/Projects/CS/vision/bsds/BSDS300-images.tgz"
)
os.system("tar -xzvf BSDS300-images.tgz")
imgs = glob.glob("BSDS300/images/test/*.jpg")
for n, i in enumerate(imgs):
f = i.split('/')[-1].split('.')[0]
print n, f
k = mmreadgray(i)
size = k.shape
scale = 4000.0 / min(size)
newsize = (int(size[0] * scale), int(size[1] * scale))
img = (mmhmin(k, 50) > 128).astype(numpy.uint8) * 255
img = mmareaclose(img, 20)
img = mmareaopen(img, 20)
pil_img = Image.fromarray(img, "L")
pil_img = pil_img.resize(newsize)
pil_img.save("output/" + f + ".png")
os.system("convert -compress none %s %s" %
("output/" + f + ".png", "output/" + f + ".pgm"))
import sys
sys.exit(0)
import glob
import Image
import os
os.system("wget http://www.eecs.berkeley.edu/Research/Projects/CS/vision/bsds/BSDS300-images.tgz")
os.system("tar -xzvf BSDS300-images.tgz")
imgs = glob.glob("BSDS300/images/test/*.jpg")
for n,i in enumerate(imgs):
f = i.split('/')[-1].split('.')[0]
print n, f
k = mmreadgray(i)
size = k.shape
scale = 4000.0/min(size)
newsize = (int(size[0]*scale), int(size[1]*scale))
img = (mmhmin(k, 50) > 128).astype(numpy.uint8)*255
img = mmareaclose(img, 20)
img = mmareaopen(img, 20)
pil_img = Image.fromarray(img, "L")
pil_img = pil_img.resize(newsize)
pil_img.save( "output/"+f+".png" )
os.system("convert -compress none %s %s" %
("output/"+f+".png", "output/"+f+".pgm"))
import sys
sys.exit(0)
