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)