def main_2(): def make_outdir(img_filename): # split a png filename into an appropriate output path pathname, filename = os.path.split(img_filename) # print pathname # print filename dirlist = pathname.split(os.sep) # print dirlist, dirlist[1:] outdir = os.path.join(*dirlist[1:]) # print outdir return outdir output_dir_base = "600_winder_{0}".format(runseg.segmentation_algorithm) input_dirname = "600_winder" img_filelist, xml_filelist = load_image_and_xml_list(input_dirname) output_dir_hash = {} for i, x in zip(img_filelist, xml_filelist): od = make_outdir(i) if not od in output_dir_hash: output_dir_hash[od] = [] output_dir_hash[od].append(get_basename(i)) pklname = "{0}_output_dir_hash.pkl".format(input_dirname) f = open(pklname, "wb") pickle.dump(output_dir_hash, f) f.close() # print "wrote",pklname # fname_fmt = "{0}_{1}".format( input_dirname for dirname in output_dir_hash.keys(): filelist = [ os.path.join(input_dirname, dirname, f) for f in output_dir_hash[dirname] ] img_filelist = [s + ".png" for s in filelist] xml_filelist = [s + ".xml" for s in filelist] output_basename = os.path.split(dirname)[-1] print img_filelist[0] print xml_filelist[0] output_dir = os.path.join(output_dir_base, dirname) print "outputdir=", output_dir print output_basename if not os.path.exists(output_dir): os.makedirs(output_dir) runseg.run_file_list(img_filelist, xml_filelist, output_dir, output_basename) sys.exit(0)
def main_2() : def make_outdir(img_filename) : # split a png filename into an appropriate output path pathname,filename = os.path.split(img_filename) # print pathname # print filename dirlist = pathname.split(os.sep) # print dirlist, dirlist[1:] outdir = os.path.join(*dirlist[1:]) # print outdir return outdir output_dir_base = "600_winder_{0}".format(runseg.segmentation_algorithm) input_dirname = "600_winder" img_filelist, xml_filelist = load_image_and_xml_list( input_dirname ) output_dir_hash = {} for i,x in zip(img_filelist,xml_filelist) : od = make_outdir(i) if not od in output_dir_hash : output_dir_hash[od] = [] output_dir_hash[od].append( get_basename(i) ) pklname = "{0}_output_dir_hash.pkl".format(input_dirname) f = open( pklname, "wb" ) pickle.dump(output_dir_hash,f) f.close() # print "wrote",pklname # fname_fmt = "{0}_{1}".format( input_dirname for dirname in output_dir_hash.keys() : filelist = [ os.path.join(input_dirname,dirname,f) for f in output_dir_hash[dirname] ] img_filelist = [ s+".png" for s in filelist ] xml_filelist = [ s+".xml" for s in filelist ] output_basename = os.path.split(dirname)[-1] print img_filelist[0] print xml_filelist[0] output_dir = os.path.join(output_dir_base,dirname) print "outputdir=",output_dir print output_basename if not os.path.exists(output_dir) : os.makedirs(output_dir) runseg.run_file_list( img_filelist, xml_filelist, output_dir, output_basename ) sys.exit(0)
def foo(): for imgdir in image_dirs: print "imgdir=", imgdir # did we land in a dir with png files? if "png" not in imgdir: continue print root, rdir, imgdir imgpath = os.path.join(root, rdir, "300dpi", imgdir) print "imgpath=", imgpath # subversion metadata keeps messing with me assert ".svn" not in rdir, rdir assert ".svn" not in imgdir, imgdir assert ".svn" not in root, root assert ".svn" not in imgpath, imgpath # did we land in a dir with png files? pngfiles = glob.glob(os.path.join(imgpath, "*.png")) assert len(pngfiles) # write the results to a path different than the source output_dir = os.path.join(imgpath) # output_dir = output_dir.replace( "300_winder/", "300_winder_vor/" ) # output_dir = output_dir.replace( "300_winder/", "300_winder_rast/" ) # output_dir = os.path.join( "300_winder_fullpage_vor", output_dir ) output_dir = os.path.join("300_winder_fullpage_rast", output_dir) # destination doesn't exist, make it if not os.path.exists(output_dir): os.makedirs(output_dir) img_filelist = pngfiles xml_filelist = [s.replace(".png", ".xml") for s in img_filelist] # output data file name is based on the last dir in the tree # print "imgpath=",imgpath output_basename = os.path.split(imgpath)[-1] # print "output_dir=",output_dir # print "output_basename=",output_basename runseg.run_file_list(img_filelist, xml_filelist, output_dir, output_basename)
def foo() : for imgdir in image_dirs : print "imgdir=",imgdir # did we land in a dir with png files? if "png" not in imgdir : continue print root,rdir,imgdir imgpath = os.path.join(root,rdir,"300dpi",imgdir) print "imgpath=",imgpath # subversion metadata keeps messing with me assert ".svn" not in rdir, rdir assert ".svn" not in imgdir, imgdir assert ".svn" not in root, root assert ".svn" not in imgpath, imgpath # did we land in a dir with png files? pngfiles = glob.glob(os.path.join(imgpath,"*.png")) assert len(pngfiles) # write the results to a path different than the source output_dir = os.path.join(imgpath) # output_dir = output_dir.replace( "300_winder/", "300_winder_vor/" ) # output_dir = output_dir.replace( "300_winder/", "300_winder_rast/" ) # output_dir = os.path.join( "300_winder_fullpage_vor", output_dir ) output_dir = os.path.join( "300_winder_fullpage_rast", output_dir ) # destination doesn't exist, make it if not os.path.exists(output_dir) : os.makedirs( output_dir ) img_filelist = pngfiles xml_filelist = [ s.replace(".png",".xml") for s in img_filelist ] # output data file name is based on the last dir in the tree # print "imgpath=",imgpath output_basename = os.path.split(imgpath)[-1] # print "output_dir=",output_dir # print "output_basename=",output_basename runseg.run_file_list( img_filelist, xml_filelist, output_dir, output_basename )