示例#1
0
文件: heal.py 项目: jp3477/whisk
        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)
示例#2
0
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
示例#3
0
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
示例#4
0
文件: heal.py 项目: chexenia/whisk
    # 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)