image_loader.lua

require 'image'
require 'paths'
require 'groundtruth_opticalflow'

ImageLoader = {}

function ImageLoader:init(geometry, path, fst, delta)
   self.geometry = geometry
   self.nextFrame = fst-1
   self.delta = delta
   self.path = path
   if self.path:sub(-1) ~= '/' then
      self.path = self.path .. '/'
   end
end

function ImageLoader:getNextFrame()
   self.nextFrame = self.nextFrame + 1
   print(self.nextFrame)
   local impath = string.format("%simages/%09d.jpg", self.path, self.nextFrame)
   if not paths.filep(impath) then
      impath = string.format("%simages/%09d.png", self.path, self.nextFrame)
      if not paths.filep(impath) then
	 return nil
      end
   end
   return image.load(impath)
   --local im = image.scale(image.load(impath), self.geometry.wImg, self.geometry.hImg)
   --return im
end

function ImageLoader:getCurrentGT()
   if self.nextFrame >= 1 then
      local pimpath = string.format("%09d",self.nextFrame-1)
      local impath = string.format("%09d", self.nextFrame)
      local _, gt = loadImageOpticalFlow(self.geometry, self.path, impath, pimpath, 1,
					 'liu')
					 --'cross-correlation')
      return gt
   else
      return torch.Tensor(2, self.geometry.wImg, self.geometry.hImg)
   end
end