utility/changeScript_v1.lua at master · AdithyaReddy9/utility · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
require 'torch'
require 'nn'
require 'optim'

local dbg = require 'debugger'


local cmd = torch.CmdLine()
cmd:option('-ipfile', 'train_01.lst_bkp') -- necessary
cmd:option('-ratio', 1) -- Neceesary to downsample a video
cmd:option('-seqLength', 16)
cmd:option('-basePath', '/home/adithya/ucf101/UCF-101/')

local opt = cmd:parse(arg)

-- Torch cmd parses user input as strings so we need to convert number strings to numbers
for k, v in pairs(opt) do
  if tonumber(v) then
    opt[k] = tonumber(v)
  end
end

function mysplit(inputstr, sep)
        if sep == nil then
                sep = "%s"
        end
        local t={} ; i=1
        for str in string.gmatch(inputstr, "([^"..sep.."]+)") do
                t[i] = str
                i = i + 1
        end
        return t
end


ft = io.open(opt.ipfile, "r")
ot = io.open("train_01.lst", "w")

line = ft:read()


while line do

st = mysplit(line, " ")
vid = st[2] -1;
st1 = mysplit(st[1], "/")
dir = opt.basePath .. st1[7] .. "/" ..  st1[8] .. ".avi"
stri = dir .. " " ..  vid .. " " .. st[3]
ot:write(stri)
ot:write('\n')

line = ft:read()
end

ot:close()
ft:close()