KPCSH/load_data.m at main · hutt94/KPCSH · 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
function [XTrain,YTrain,LTrain,XTest,YTest,LTest] = load_data(db_name)

load(['./datasets/',db_name,'.mat']);
rng(2023);

if strcmp(db_name, 'IAPRTC-12')
    clear V_tr V_te
    X = [I_tr; I_te]; Y = [T_tr; T_te]; L = [L_tr; L_te];
    R = randperm(size(L,1));
    queryInds = R(1:2000);
    sampleInds = R(2001:end);
    XTrain = X(sampleInds, :); YTrain = Y(sampleInds, :); LTrain = L(sampleInds, :);
    XTest = X(queryInds, :); YTest = Y(queryInds, :); LTest = L(queryInds, :);

elseif strcmp(db_name, 'MIRFLICKR')
    X = [I_tr; I_te]; Y = [T_tr; T_te]; L = [L_tr; L_te];
    R = randperm(size(L,1));
    queryInds = R(1:2000);
    sampleInds = R(2001:end);
    XTrain = X(sampleInds, :); YTrain = Y(sampleInds, :); LTrain = L(sampleInds, :);
    XTest = X(queryInds, :); YTest = Y(queryInds, :); LTest = L(queryInds, :);

elseif strcmp(db_name, 'NUSWIDE10')
    X = [I_tr; I_te]; Y = [T_tr; T_te]; L = [L_tr; L_te];
    R = randperm(size(L,1));
    queryInds = R(1:2000);
    sampleInds = R(2001:end);
    XTrain = X(sampleInds, :); YTrain = Y(sampleInds, :); LTrain = L(sampleInds, :);
    XTest = X(queryInds, :); YTest = Y(queryInds, :); LTest = L(queryInds, :);
    clear I_tr I_te T_tr T_te L_tr L_te

elseif strcmp(db_name, 'MIRFLICKR_deep')
    X = (X-min(min(X)))/(max(max(X))-min(min(X)));
    R = randperm(size(X,1));
    queryInds = R(1:2000);
    sampleInds = R(2001:end);
    XTrain = X(sampleInds, :); YTrain = Y(sampleInds, :); LTrain = L(sampleInds, :);
    XTest = X(queryInds, :); YTest = Y(queryInds, :); LTest = L(queryInds, :);
end
clear X Y L PCA_Y R XAll


end