TFNH/setting.py at master · hutt94/TFNH · 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
import scipy.io as sio

# WIKI
dataset = 'data/WikiData.mat'

# batch_size
pair = True
pair_batch_size = 217
img_batch_size = 217
txt_batch_size = 217
BATCH_NUM = 10
batch_size = pair_batch_size + img_batch_size + txt_batch_size

# epoch
TOTAL_EPOCH = 200
kp = 0.5

# network structures setting
hid_dim = 1024
hash_dim = 64
dis_dim = 64
dis_out_dim = 2

# hyper-parameters setting
alpha = 1
beta = [0.1, 0.3, 0.3]
mu = 2
yita = [20, 20]
lr_fn = 0.001
lr_dn = 0.001

# load data
data = sio.loadmat(dataset)
I_tr = data['I_tr']
T_tr = data['T_tr']
L_tr = data['L_tr']
I_te = data['I_te']
T_te = data['T_te']
L_te = data['L_te']
del data

# input dimension
img_dim = I_tr.shape[1]
txt_dim = T_tr.shape[1]
lab_dim = L_tr.shape[1]
all_num = L_tr.shape[0]
fusion_dim = img_dim + txt_dim