From c85491111758e459a42fe9c903f26d990043582c Mon Sep 17 00:00:00 2001 From: Adrian Bulat Date: Mon, 11 Sep 2017 13:37:10 +0100 Subject: [PATCH] Adapt txt format to 3D --- main.lua | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/main.lua b/main.lua index 33e43eb..783804a 100644 --- a/main.lua +++ b/main.lua @@ -100,7 +100,11 @@ for i = 1, #fileList do -- csv without header local out = torch.DiskFile(dest .. '.txt', 'w') for i=1,68 do - out:writeString(tostring(preds_img[{1,i,1}]) .. ',' .. tostring(preds_img[{1,i,2}]) .. '\n') + if preds_img:size(2)==3 then + out:writeString(tostring(preds_img[{i,1}]) .. ',' .. tostring(preds_img[{i,2}]) .. ',' .. tostring(preds_img[{i,3}]) .. '\n') + else + out:writeString(tostring(preds_img[{i,1}]) .. ',' .. tostring(preds_img[{i,2}]) .. '\n') + end end out:close() end