-
Notifications
You must be signed in to change notification settings - Fork 3
/
data.lua
35 lines (31 loc) · 1.48 KB
/
data.lua
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
paths.dofile('io/readFiles.lua')
paths.dofile('io/readGEFiles.lua');
paths.dofile('io/compute_mean_std.lua');
paths.dofile('io/readBatch.lua');
function GetASiameseBatch(nPositiveImages, nDifferentAngleImages, nDifferentCategoryImages, opt)
local status, input, target = coroutine.resume(GetASiameseBatchCoroutine,
nPositiveImages, nDifferentAngleImages, nDifferentCategoryImages, opt.test);
return input, target
end
function GetAnImageBatch(batchSize, opt)
local status, input, target = coroutine.resume(GetAnImageBatchCoroutine,
batchSize, opt.viewpoint, opt.test,
opt.deterministic, opt.spline);
return input, target
end
function GetAUniformImageBatch(batchSize, opt)
local status, input, target = coroutine.resume(GetAUniformImageBatchCoroutine,
batchSize, opt.viewpoint, opt.test,
opt.spline);
return input, target
end
function GetAUniformAnimationBatch(batchSize, opt)
local status, input, target = coroutine.resume(GetAUniformAnimationBatchCoroutine,
batchSize, opt.viewpoint, opt.spline);
return input, target
end
function GetAVideoBatch(opt)
local status, input, target = coroutine.resume(GetAVideoBatchCoroutine,
opt.viewpoint, opt.test, opt.spline);
return input, target
end