require 'torch';

require 'nn';

require 'image';

cuda = false

--if cuda then

    require 'cutorch';

    require 'cunn';

--end

require 'sys';

local dir = require 'pl.dir';

local c = os.clock()

local t = os.time()

local trainFolder = '/home/andrew/mitosis/MITOS/training/'

local mapFolder = '/home/andrew/mitosis/maps/'

local netPath = '/home/andrew/mitosis/nets/net.t7'

dofile("getImagePaths.lua")

imagePaths = getImagePaths(trainFolder)

if paths.dirp(mapFolder) == false then

    paths.mkdir(mapFolder)

end

dofile("scan.lua")

dofile("expand.lua")

for k,imagePath in ipairs(imagePaths) do

    print(k)

    local c1 = os.clock()

    local t1 = os.time()

    local net = torch.load(netPath)

    net = expand(net)

    if cuda then

        net = net:cuda()

    else

        net = net:float()

    end

    local img = image.load(imagePath, 3, 'float')

    if cuda then

        img = img:cuda()

    else

        img = img:float()

    end

    local map = scan(img, net)

    local outfile = paths.concat(mapFolder, paths.basename(imagePath))

    image.save(outfile, map)

    print(os.clock()-c1)

    print(os.time()-t1)

end

print(os.clock()-c)

print(os.time()-t)