openface/training/opts.lua

local lfs = require 'lfs'

local M = { }

-- http://stackoverflow.com/questions/6380820/get-containing-path-of-lua-file
function script_path()
   local str = debug.getinfo(2, "S").source:sub(2)
   return str:match("(.*/)")
end

function M.parse(arg)

   local cmd = torch.CmdLine()
   cmd:text()
   cmd:text('OpenFace')
   cmd:text()
   cmd:text('Options:')

   ------------ General options --------------------
   cmd:option('-cache',
              paths.concat(script_path(), 'work'),
              'subdirectory in which to save/log experiments')
   cmd:option('-data',
              paths.concat(os.getenv('HOME'), 'openface', 'data',
                           'casia-facescrub',
                           'dlib-affine-sz:96'),
                           -- 'dlib-affine-224-split'),
              'Home of dataset. Split into "train" and "val" directories that separate images by class.')
   cmd:option('-manualSeed',         2, 'Manually set RNG seed')

   ------------- Data options ------------------------
   cmd:option('-nDonkeys',        2, 'number of donkeys to initialize (data loading threads)')

   ------------- Training options --------------------
   cmd:option('-nEpochs',         1000, 'Number of total epochs to run')
   cmd:option('-epochSize',       1000, 'Number of batches per epoch')
   cmd:option('-testEpochSize',   300,  'Number of batches to test per epoch')
   cmd:option('-epochNumber',     1,    'Manual epoch number (useful on restarts)')
   cmd:option('-peoplePerBatch',  45,   'Number of people to sample in each mini-batch.')
   cmd:option('-imagesPerPerson', 40,   'Number of images to sample per person in each mini-batch.')
   cmd:option('-batchSize',   100,   'Minibatch size')

   ---------- Model options ----------------------------------
   cmd:option('-retrain',     'none', 'provide path to model to retrain with')
   cmd:option('-modelDef', '../models/openface/nn4.def.lua', 'path to model definiton')
   cmd:option('-imgDim', 96, 'Image dimension. nn2=224, nn4=96')
   cmd:text()

   local opt = cmd:parse(arg or {})
   os.execute('mkdir -p ' .. opt.cache)
   local count = 1
   for f in lfs.dir(opt.cache) do
      local isDir = paths.dirp(paths.concat(opt.cache, f))
      if f ~= "." and f ~= ".." and isDir then
         count = count + 1
      end
   end
   opt.save = paths.concat(opt.cache, count)

   return opt
end

return M