def load_episode(self, index, data_type='train'):
"""The function to load the episodes.
Args:
index: the index for the episodes.
data_type: the phase for meta-learning.
"""
if data_type == 'train':
data_list = self.train_data
epite_sample_num = FLAGS.metatrain_epite_sample_num
elif data_type == 'test':
data_list = self.test_data
if FLAGS.metatest_epite_sample_num == 0:
epite_sample_num = FLAGS.shot_num
else:
epite_sample_num = FLAGS.metatest_episode_test_sample
elif data_type == 'val':
data_list = self.val_data
if FLAGS.metatest_epite_sample_num == 0:
epite_sample_num = FLAGS.shot_num
else:
epite_sample_num = FLAGS.metatest_episode_test_sample
else:
raise Exception('Please check data list type')
dim_input = FLAGS.img_size * FLAGS.img_size * 3
epitr_sample_num = FLAGS.shot_num
this_episode = data_list[index]
this_task_tr_filenames = this_episode['filenamea']
this_task_te_filenames = this_episode['filenameb']
this_task_tr_labels = this_episode['labela']
this_task_te_labels = this_episode['labelb']
'''
if FLAGS.metatrain is False and FLAGS.base_augmentation:
this_inputa, this_labela = process_batch_augmentation(this_task_tr_filenames, \
this_task_tr_labels, dim_input, epitr_sample_num)
this_inputb, this_labelb = process_batch(this_task_te_filenames, \
this_task_te_labels, dim_input, epite_sample_num)
else:
'''
this_inputa, this_labela = process_batch(this_task_tr_filenames, \
this_task_tr_labels, dim_input, epitr_sample_num)
this_inputb, this_labelb = process_batch(this_task_te_filenames, \
this_task_te_labels, dim_input, epite_sample_num)
return this_inputa, this_labela, this_inputb, this_labelb
def load_episode(self, index, data_type='train'):
if data_type == 'train':
data_list = self.train_data
epite_sample_num = FLAGS.metatrain_epite_sample_num
elif data_type == 'test':
data_list = self.test_data
if FLAGS.metatest_epite_sample_num == 0:
epite_sample_num = FLAGS.shot_num
else:
epite_sample_num = FLAGS.metatest_episode_test_sample
elif data_type == 'val':
data_list = self.val_data
if FLAGS.metatest_epite_sample_num == 0:
epite_sample_num = FLAGS.shot_num
else:
epite_sample_num = FLAGS.metatest_episode_test_sample
else:
print('[Error] Please check data list type')
dim_input = FLAGS.img_size * FLAGS.img_size * 3
epitr_sample_num = FLAGS.shot_num
this_episode = data_list[index]
this_task_tr_filenames = this_episode['filenamea']
this_task_te_filenames = this_episode['filenameb']
this_task_tr_labels = this_episode['labela']
this_task_te_labels = this_episode['labelb']
if FLAGS.base_augmentation and FLAGS.metatrain == False:
this_inputa, this_labela = process_batch_augmentation(this_task_tr_filenames, \
this_task_tr_labels, dim_input, epitr_sample_num)
this_inputb, this_labelb = process_batch(this_task_te_filenames, \
this_task_te_labels, dim_input, epite_sample_num)
else:
this_inputa, this_labela = process_batch(this_task_tr_filenames, \
this_task_tr_labels, dim_input, epitr_sample_num)
this_inputb, this_labelb = process_batch(this_task_te_filenames, \
this_task_te_labels, dim_input, epite_sample_num)
return this_inputa, this_labela, this_inputb, this_labelb
def test_with_d(self):
NUM_TEST_POINTS = 600
np.random.seed(1)
metaval_accuracies = []
metaval_recur_accuracies = []
num_samples_per_class = FLAGS.shot_num * 2
task_num = FLAGS.way_num * num_samples_per_class
task_un_num = 10 * FLAGS.way_num
task_dis_num = FLAGS.nb_ul_samples
half_num_samples = FLAGS.shot_num
dim_input = FLAGS.img_size * FLAGS.img_size * 3
if FLAGS.dataset == 'miniImagenet':
data_name = 'mini_'
else:
data_name = 'tiered_'
####
filename_dir = FLAGS.logdir_base + '/' + 'filenames_and_labels/'
this_setting_filename_dir = filename_dir + data_name + str(
FLAGS.shot_num) + 'shot_' + str(FLAGS.way_num) + 'way/'
all_filenames = np.load(this_setting_filename_dir +
'test_filenames.npy').tolist()
labels = np.load(this_setting_filename_dir +
'test_labels.npy').tolist()
un_file_dict = {}
un_label_dict = {}
####
un_new_folder = this_setting_filename_dir + 'unlabeled_samples_for_test'
for f in range(self.unfiles_num):
un_file_dict[f] = np.load(un_new_folder + '/' + 'test_' + str(f) +
'_un_filenames.npy').tolist()
un_label_dict[f] = np.load(un_new_folder + '/' + 'test_' + str(f) +
'_un_labels.npy').tolist()
dis_file_dict = {}
dis_folder = un_new_folder + '/distracting_file'
for n in range(FLAGS.num_dis):
dis_file_dict[n] = np.load(dis_folder + '/' + 'test_' +
str(n + 1) +
'_dis_filenames.npy').tolist()
for test_idx in trange(NUM_TEST_POINTS):
this_task_filenames = all_filenames[test_idx *
task_num:(test_idx + 1) *
task_num]
this_task_un_filenames = []
this_task_un_labels = []
for j in range(self.unfiles_num):
this_task_un_filenames.extend(
un_file_dict[j][test_idx * task_un_num:(test_idx + 1) *
task_un_num])
this_task_un_labels.extend(
un_label_dict[j][test_idx * task_un_num:(test_idx + 1) *
task_un_num])
this_task_dis_filenames = []
for k in range(FLAGS.num_dis):
current_dis_files = dis_file_dict[k][test_idx *
100:(test_idx + 1) * 100]
this_task_dis_filenames.extend(
current_dis_files[:task_dis_num])
this_task_tr_filenames = []
this_task_tr_labels = []
this_task_te_filenames = []
this_task_te_labels = []
for class_k in range(FLAGS.way_num):
this_class_filenames = this_task_filenames[
class_k * num_samples_per_class:(class_k + 1) *
num_samples_per_class]
this_class_label = labels[class_k *
num_samples_per_class:(class_k + 1) *
num_samples_per_class]
this_task_tr_filenames += this_class_filenames[
0:half_num_samples]
this_task_tr_labels += this_class_label[0:half_num_samples]
this_task_te_filenames += this_class_filenames[
half_num_samples:]
this_task_te_labels += this_class_label[half_num_samples:]
inputa, labela = process_batch(this_task_tr_filenames,
this_task_tr_labels,
dim_input,
half_num_samples,
reshape_with_one=True)
inputb, labelb = process_batch(this_task_te_filenames,
this_task_te_labels,
dim_input,
half_num_samples,
reshape_with_one=True)
inputc, labelc = process_un_batch_recur(this_task_un_filenames,
this_task_un_labels,
dim_input,
reshape_with_one=True)
inputd = process_dis_batch_recur(this_task_dis_filenames,
dim_input,
num=task_dis_num * FLAGS.num_dis,
reshape_with_one=True)
feed_dict = {
self.model.inputa: inputa,
self.model.inputb: inputb,
self.model.labela: labela,
self.model.labelb: labelb,
self.model.inputc: inputc,
self.model.labelc: labelc,
self.model.inputd: inputd,
self.model.meta_lr: 0.0
}
result = self.sess.run([
self.model.metaval_total_accuracies,
self.model.metaval_total_recur_accuracies
], feed_dict)
metaval_accuracies.append(result[0])
metaval_recur_accuracies.append(result[1])
if test_idx % 600 == 0:
tmp_accies = np.array(metaval_accuracies)
tmp_means = np.mean(tmp_accies, 0)
max_acc = np.max(tmp_means)
print('***** Best Acc: ' + str(max_acc))
metaval_accuracies = np.array(metaval_accuracies)
metaval_recur_accuracies = np.array(metaval_recur_accuracies)
means = np.mean(metaval_accuracies, 0)
recur_means = np.mean(metaval_recur_accuracies, 0)
max_idx = np.argmax(means)
max_acc = np.max(means)
stds = np.std(metaval_accuracies, 0)
ci95 = 1.96 * stds / np.sqrt(NUM_TEST_POINTS)
max_ci95 = ci95[max_idx]
print('Mean validation accuracy and confidence intervals')
print((means, ci95))
print('***** Best Acc: ' + str(max_acc) + ' CI95: ' + str(max_ci95))
#####
logdir_base = FLAGS.test_output_dir + '/' + FLAGS.exp_string
if not os.path.exists(logdir_base):
os.mkdir(logdir_base)
logdir = logdir_base + '/' + 'logdir_d'
if not os.path.exists(logdir):
os.mkdir(logdir)
out_filename = logdir + '/' + 'result_noaug_' + str(
FLAGS.shot_num) + 'shot_' + str(
FLAGS.test_iter) + 'base_unlabeled_' + str(
FLAGS.nb_ul_samples) + '_recur.csv'
out_pkl = logdir + '/' + 'result_noaug_' + str(
FLAGS.shot_num) + 'shot_' + str(
FLAGS.test_iter) + 'base_unlabeled_' + str(
FLAGS.nb_ul_samples) + '_recur.pkl'
with open(out_pkl, 'wb') as f:
pickle.dump({'mses': metaval_accuracies}, f)
with open(out_filename, 'w') as f:
writer = csv.writer(f, delimiter=',')
writer.writerow(['update' + str(i) for i in range(len(means))])
writer.writerow(means)
writer.writerow(recur_means)
writer.writerow(stds)
writer.writerow(ci95)
def train(self):
exp_string = FLAGS.exp_string
train_writer = tf.summary.FileWriter(FLAGS.logdir + '/' + exp_string,
self.sess.graph)
print('Done initializing, starting training')
loss_list, acc_list, pre_train_loss_list = [], [], []
train_lr = FLAGS.meta_lr
swn_lr = FLAGS.swn_lr
task_un_num = FLAGS.nb_ul_samples * FLAGS.way_num
num_samples_per_class = FLAGS.shot_num + 15
task_num = FLAGS.way_num * num_samples_per_class
num_samples_per_class_test = FLAGS.shot_num * 2
test_task_num = FLAGS.way_num * num_samples_per_class_test
epitr_sample_num = FLAGS.shot_num
epite_sample_num = 15
test_task_sample_num = FLAGS.shot_num
dim_input = FLAGS.img_size * FLAGS.img_size * 3
filename_dir = FLAGS.logdir_base + '/' + 'filenames_and_labels/'
if FLAGS.dataset == 'miniImagenet':
data_name = 'mini_'
else:
data_name = 'tiered_'
this_setting_filename_dir = filename_dir + data_name + str(
FLAGS.shot_num) + 'shot_' + str(FLAGS.way_num) + 'way/'
all_filenames = np.load(this_setting_filename_dir +
'train_filenames.npy').tolist()
labels = np.load(this_setting_filename_dir +
'train_labels.npy').tolist()
un_train_folder = this_setting_filename_dir + 'unlabeled_samples_for_train/'
all_unlabeled_filenames = np.load(un_train_folder + 'train' + '_' +
str(FLAGS.nb_ul_samples) +
'_un_filenames.npy').tolist()
all_unlabeled_labels = np.load(un_train_folder + 'train' + '_' +
str(FLAGS.nb_ul_samples) +
'_un_labels.npy').tolist()
all_test_filenames = np.load(this_setting_filename_dir +
'val_filenames.npy').tolist()
test_labels = np.load(this_setting_filename_dir +
'val_labels.npy').tolist()
un_val_folder = this_setting_filename_dir + 'unlabeled_samples_for_val/'
test_unlabeled_filenames = np.load(un_val_folder + 'val' + '_' +
str(FLAGS.nb_ul_samples) +
'_un_filenames.npy').tolist()
test_unlabeled_labels = np.load(un_val_folder + 'val' + '_' +
str(FLAGS.nb_ul_samples) +
'_un_labels.npy').tolist()
test_idx = 0
for train_idx in trange(FLAGS.metatrain_iterations):
inputa = []
labela = []
inputb = []
labelb = []
inputc = []
labelc = []
for meta_batch_idx in range(FLAGS.meta_batch_size):
this_task_filenames = all_filenames[
(train_idx * FLAGS.meta_batch_size + meta_batch_idx) *
task_num:(train_idx * FLAGS.meta_batch_size +
meta_batch_idx + 1) * task_num]
this_task_un_filenames = all_unlabeled_filenames[
(train_idx * FLAGS.meta_batch_size + meta_batch_idx) *
task_un_num:(train_idx * FLAGS.meta_batch_size +
meta_batch_idx + 1) * task_un_num]
this_task_un_labels = all_unlabeled_labels[
(train_idx * FLAGS.meta_batch_size + meta_batch_idx) *
task_un_num:(train_idx * FLAGS.meta_batch_size +
meta_batch_idx + 1) * task_un_num]
this_task_tr_filenames = []
this_task_tr_labels = []
this_task_te_filenames = []
this_task_te_labels = []
for class_k in range(FLAGS.way_num):
this_class_filenames = this_task_filenames[
class_k * num_samples_per_class:(class_k + 1) *
num_samples_per_class]
this_class_label = labels[class_k *
num_samples_per_class:(class_k +
1) *
num_samples_per_class]
this_task_tr_filenames += this_class_filenames[
0:epitr_sample_num]
this_task_tr_labels += this_class_label[0:epitr_sample_num]
this_task_te_filenames += this_class_filenames[
epitr_sample_num:]
this_task_te_labels += this_class_label[epitr_sample_num:]
this_inputa, this_labela = process_batch(
this_task_tr_filenames,
this_task_tr_labels,
dim_input,
epitr_sample_num,
reshape_with_one=False)
this_inputb, this_labelb = process_batch(
this_task_te_filenames,
this_task_te_labels,
dim_input,
epite_sample_num,
reshape_with_one=False)
this_inputc, this_labelc = process_un_batch(
this_task_un_filenames, this_task_un_labels, dim_input)
inputa.append(this_inputa)
labela.append(this_labela)
inputb.append(this_inputb)
labelb.append(this_labelb)
inputc.append(this_inputc)
labelc.append(this_labelc)
inputa = np.array(inputa)
labela = np.array(labela)
inputb = np.array(inputb)
labelb = np.array(labelb)
inputc = np.array(inputc)
labelc = np.array(labelc)
feed_dict = {
self.model.inputa: inputa,
self.model.inputb: inputb,
self.model.labela: labela,
self.model.labelb: labelb,
self.model.inputc: inputc,
self.model.labelc: labelc,
self.model.meta_lr: train_lr,
self.model.swn_lr: swn_lr
}
input_tensors = [self.model.metatrain_op]
input_tensors.extend([self.model.meta_swn_train_op])
input_tensors.extend([self.model.final_pretrain_loss])
input_tensors.extend([self.model.total_loss])
input_tensors.extend([self.model.total_accuracy])
if (train_idx % FLAGS.meta_sum_step == 0
or train_idx % FLAGS.meta_print_step == 0):
input_tensors.extend(
[self.model.summ_op, self.model.total_loss])
result = self.sess.run(input_tensors, feed_dict)
loss_list.append(result[3])
acc_list.append(result[4])
pre_train_loss_list.append(result[2])
if train_idx % FLAGS.meta_sum_step == 0:
train_writer.add_summary(result[5], train_idx)
if train_idx % FLAGS.meta_print_step == 0:
print_str = 'Iteration:' + str(train_idx)
print_str += ' Loss:' + str(
np.mean(loss_list)) + ' Acc:' + str(np.mean(acc_list))
print_str += ' Pre Loss:' + str(np.mean(pre_train_loss_list))
print(print_str)
loss_list, acc_list, pre_train_loss_list = [], [], []
if (train_idx != 0) and train_idx % FLAGS.meta_save_step == 0:
weights = self.sess.run(self.model.weights)
ss_weights = self.sess.run(self.model.ss_weights)
fc_weights = self.sess.run(self.model.fc_weights)
swn_weights = self.sess.run(self.model.swn_weights)
np.save(
FLAGS.logdir + '/' + exp_string + '/weights_' +
str(train_idx) + '.npy', weights)
np.save(
FLAGS.logdir + '/' + exp_string + '/ss_weights_' +
str(train_idx) + '.npy', ss_weights)
np.save(
FLAGS.logdir + '/' + exp_string + '/fc_weights_' +
str(train_idx) + '.npy', fc_weights)
np.save(
FLAGS.logdir + '/' + exp_string + '/swn_weights_' +
str(train_idx) + '.npy', swn_weights)
if train_idx % FLAGS.meta_val_print_step == 0:
test_loss = []
test_accs = []
test_swn_loss = []
for test_itr in range(10):
this_test_task_filenames = all_test_filenames[
test_idx * test_task_num:(test_idx + 1) *
test_task_num]
this_test_task_un_filenames = test_unlabeled_filenames[
test_idx * task_un_num:(test_idx + 1) * task_un_num]
this_test_task_un_labels = test_unlabeled_labels[
test_idx * task_un_num:(test_idx + 1) * task_un_num]
this_test_task_tr_filenames = []
this_test_task_tr_labels = []
this_test_task_te_filenames = []
this_test_task_te_labels = []
for class_k in range(FLAGS.way_num):
this_test_class_filenames = this_test_task_filenames[
class_k *
num_samples_per_class_test:(class_k + 1) *
num_samples_per_class_test]
this_test_class_label = test_labels[
class_k *
num_samples_per_class_test:(class_k + 1) *
num_samples_per_class_test]
this_test_task_tr_filenames += this_test_class_filenames[
0:test_task_sample_num]
this_test_task_tr_labels += this_test_class_label[
0:test_task_sample_num]
this_test_task_te_filenames += this_test_class_filenames[
test_task_sample_num:]
this_test_task_te_labels += this_test_class_label[
test_task_sample_num:]
test_inputa, test_labela = process_batch(
this_test_task_tr_filenames,
this_test_task_tr_labels,
dim_input,
test_task_sample_num,
reshape_with_one=True)
test_inputb, test_labelb = process_batch(
this_test_task_te_filenames,
this_test_task_te_labels,
dim_input,
test_task_sample_num,
reshape_with_one=True)
test_inputc, test_labelc = process_un_batch(
this_test_task_un_filenames,
this_test_task_un_labels,
dim_input,
reshape_with_one=True)
test_feed_dict = {
self.model.inputa: test_inputa,
self.model.inputb: test_inputb,
self.model.labela: test_labela,
self.model.labelb: test_labelb,
self.model.inputc: test_inputc,
self.model.labelc: test_labelc,
self.model.meta_lr: 0.0,
self.model.swn_lr: 0.0
}
test_input_tensors = [
self.model.final_pretrain_loss, self.model.total_loss,
self.model.total_accuracy
]
test_result = self.sess.run(test_input_tensors,
test_feed_dict)
test_swn_loss.append(test_result[0])
test_loss.append(test_result[1])
test_accs.append(test_result[2])
test_idx += 1
print_str = '[***] Val Loss:' + str(
np.mean(test_loss) *
FLAGS.meta_batch_size) + ' Val Acc:' + str(
np.mean(test_accs) *
FLAGS.meta_batch_size) + ' Val Pre Loss:' + str(
np.mean(test_swn_loss) * FLAGS.meta_batch_size)
print(print_str)
if (train_idx != 0) and train_idx % FLAGS.lr_drop_step == 0:
train_lr = train_lr * 0.5
swn_lr = swn_lr * 0.5
if train_lr < FLAGS.min_meta_lr:
train_lr = FLAGS.min_meta_lr
if swn_lr < FLAGS.min_meta_lr:
swn_lr = FLAGS.min_meta_lr
print('Train LR: {}'.format(train_lr))
print('Train SWN LR: {}'.format(swn_lr))
weights = self.sess.run(self.model.weights)
ss_weights = self.sess.run(self.model.ss_weights)
fc_weights = self.sess.run(self.model.fc_weights)
swn_weights = self.sess.run(self.model.swn_weights)
np.save(
FLAGS.logdir + '/' + exp_string + '/weights_' +
str(train_idx + 1) + '.npy', weights)
np.save(
FLAGS.logdir + '/' + exp_string + '/ss_weights_' +
str(train_idx + 1) + '.npy', ss_weights)
np.save(
FLAGS.logdir + '/' + exp_string + '/fc_weights_' +
str(train_idx + 1) + '.npy', fc_weights)
np.save(
FLAGS.logdir + '/' + exp_string + '/swn_weights_' +
str(train_idx + 1) + '.npy', swn_weights)
def test(self):
NUM_TEST_POINTS = 600
exp_string = FLAGS.exp_string
np.random.seed(1)
metaval_accuracies = []
num_samples_per_class = FLAGS.shot_num*2
task_num = FLAGS.way_num * num_samples_per_class
half_num_samples = FLAGS.shot_num
dim_input = FLAGS.img_size * FLAGS.img_size * 3
filename_dir = FLAGS.logdir_base + 'filenames_and_labels/'
this_setting_filename_dir = filename_dir + str(FLAGS.shot_num) + 'shot_' + str(FLAGS.way_num) + 'way/'
all_filenames = np.load(this_setting_filename_dir + 'test_filenames.npy').tolist()
labels = np.load(this_setting_filename_dir + 'test_labels.npy').tolist()
for test_idx in trange(NUM_TEST_POINTS):
this_task_filenames = all_filenames[test_idx*task_num:(test_idx+1)*task_num]
this_task_tr_filenames = []
this_task_tr_labels = []
this_task_te_filenames = []
this_task_te_labels = []
for class_k in range(FLAGS.way_num):
this_class_filenames = this_task_filenames[class_k*num_samples_per_class:(class_k+1)*num_samples_per_class]
this_class_label = labels[class_k*num_samples_per_class:(class_k+1)*num_samples_per_class]
this_task_tr_filenames += this_class_filenames[0:half_num_samples]
this_task_tr_labels += this_class_label[0:half_num_samples]
this_task_te_filenames += this_class_filenames[half_num_samples:]
this_task_te_labels += this_class_label[half_num_samples:]
if FLAGS.base_augmentation or FLAGS.test_base_augmentation:
inputa, labela = process_batch_augmentation(this_task_tr_filenames, this_task_tr_labels, dim_input, half_num_samples)
inputb, labelb = process_batch_augmentation(this_task_te_filenames, this_task_te_labels, dim_input, half_num_samples)
else:
inputa, labela = process_batch(this_task_tr_filenames, this_task_tr_labels, dim_input, half_num_samples)
inputb, labelb = process_batch(this_task_te_filenames, this_task_te_labels, dim_input, half_num_samples)
feed_dict = {self.model.inputa: inputa, self.model.inputb: inputb, self.model.labela: labela, self.model.labelb: labelb, self.model.meta_lr: 0.0}
result = self.sess.run(self.model.metaval_total_accuracies, feed_dict)
metaval_accuracies.append(result)
metaval_accuracies = np.array(metaval_accuracies)
means = np.mean(metaval_accuracies, 0)
max_idx = np.argmax(means)
max_acc = np.max(means)
stds = np.std(metaval_accuracies, 0)
ci95 = 1.96*stds/np.sqrt(NUM_TEST_POINTS)
max_ci95 = ci95[max_idx]
print('Mean validation accuracy and confidence intervals')
print((means, ci95))
print('***** Best Acc: '+ str(max_acc) + ' CI95: ' + str(max_ci95))
if FLAGS.base_augmentation or FLAGS.test_base_augmentation:
out_filename = FLAGS.logdir +'/'+ exp_string + '/' + 'result_aug_' + str(FLAGS.shot_num) + 'shot_' + str(FLAGS.test_iter) + '.csv'
out_pkl = FLAGS.logdir +'/'+ exp_string + '/' + 'result_aug_' + str(FLAGS.shot_num) + 'shot_' + str(FLAGS.test_iter) + '.pkl'
else:
out_filename = FLAGS.logdir +'/'+ exp_string + '/' + 'result_noaug_' + str(FLAGS.shot_num) + 'shot_' + str(FLAGS.test_iter) + '.csv'
out_pkl = FLAGS.logdir +'/'+ exp_string + '/' + 'result_noaug_' + str(FLAGS.shot_num) + 'shot_' + str(FLAGS.test_iter) + '.pkl'
with open(out_pkl, 'wb') as f:
pickle.dump({'mses': metaval_accuracies}, f)
with open(out_filename, 'w') as f:
writer = csv.writer(f, delimiter=',')
writer.writerow(['update'+str(i) for i in range(len(means))])
writer.writerow(means)
writer.writerow(stds)
writer.writerow(ci95)