--- a +++ b/code/detectorSimple.lua @@ -0,0 +1,60 @@ +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)