options.max_angle *= pi / 180.0 # convert from degrees to radians # try to be friendly about args assert (os.path.isfile(args[0])) # "Could not find %s"%args[0] ) moviename = args[0] srcname = None if len(args) < 2 else args[1] dstname = None if len(args) < 3 else args[2] getroot = lambda nm: os.path.splitext(nm)[0] gettail = lambda lbl, ext: '[%s]%s' % (lbl, ext) if lbl else ext if len(args) == 1: #guess source and dest name based on movie name root = getroot(moviename) srcname = root + gettail(options.srclabel, '.whiskers') dstname = root + gettail(options.destlabel, '.whiskers') assert (os.path.isfile(srcname)) #, "Could not find %s."%srcname ) if len(args) == 2: #guess dest name based on source name srcname = args[1] if os.path.splitext(srcname)[1] != '.whiskers': srcname += '.whiskers' assert os.path.isfile(srcname), "Could not find %s." % srcname root = getroot(srcname) dstname = root + gettail(options.destlabel, '.whiskers') print "Loading..." w, movie = load(moviename, srcname) # need to remove srclabel and destlabel options.__dict__.pop('srclabel') options.__dict__.pop('destlabel') fix(w, movie, **options.__dict__) save_whiskers(dstname, w)
def test2(): from ui.whiskerdata import save_whiskers w,movie = load() merge_all(w,movie[0].shape) save_whiskers('merge.whiskers',w) return w,movie
def test2(): from ui.whiskerdata import save_whiskers w, movie = load() merge_all(w, movie[0].shape) save_whiskers('merge.whiskers', w) return w, movie
# try to be friendly about args assert( os.path.isfile(args[0]) ) # "Could not find %s"%args[0] ) moviename = args[0] srcname = None if len(args)<2 else args[1] dstname = None if len(args)<3 else args[2] getroot = lambda nm: os.path.splitext( nm )[0] gettail = lambda lbl,ext: '[%s]%s'%(lbl,ext) if lbl else ext if len(args)==1: #guess source and dest name based on movie name root = getroot( moviename ) srcname = root + gettail(options.srclabel,'.whiskers') dstname = root + gettail(options.destlabel,'.whiskers') assert( os.path.isfile( srcname ) ) #, "Could not find %s."%srcname ) if len(args)==2: #guess dest name based on source name srcname =args[1] if os.path.splitext(srcname)[1] != '.whiskers': srcname += '.whiskers' assert os.path.isfile( srcname ) , "Could not find %s."%srcname root = getroot( srcname ) dstname = root + gettail(options.destlabel,'.whiskers') print "Loading..." w,movie = load(moviename,srcname) # need to remove srclabel and destlabel options.__dict__.pop('srclabel') options.__dict__.pop('destlabel') fix(w,movie, **options.__dict__) save_whiskers(dstname,w)