a b/code/randRotateMirror.lua
1
function randRotateMirror(dataset)
2
    for i=1,dataset:size() do
3
        theta = math.pi/2*torch.random(0,3)
4
        sample = dataset[i][1]
5
        sample = sample:double()
6
        sample = image.rotate(sample,theta)
7
        if torch.rand(1,2) == 1 then
8
            sample = image.hflip(sample)
9
        end
10
        dataset[i][1] = sample
11
    end
12
13
    return dataset
14
end