Correct method name on Middlebury.

Laulian · Jan 26, 2016 · d1876c7 · d1876c7
1 parent e3bb91c
commit d1876c7
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/main.lua b/main.lua
@@ -1203,8 +1203,9 @@ for _, i in ipairs(examples) do
          -- savePNG(('tmp/fos_%d.png'):format(i), pred)
          base = 'out/' .. fname_submit[i - (#X - #fname_submit)]
          os.execute('mkdir -p ' .. base)
-         adcensus.writePFM(image.vflip(pred[{1,1}]:float()), base .. '/disp0MC-CNN.pfm')
-         local f = io.open(base .. '/timeMC-CNN.txt', 'w')
+         local method_name = 'MC-CNN-' .. (arch == 'fast' and 'fst' or 'acrt' )
+         adcensus.writePFM(image.vflip(pred[{1,1}]:float()), base .. '/disp0' .. method_name .. '.pfm')
+         local f = io.open(base .. '/time' .. method_name .. '.txt', 'w')
          f:write(tostring(runtime))
          f:close()
       end