Тренируйте Caffe CNN для вывода многомерных функций - PullRequest
Новая
       55

Тренируйте Caffe CNN для вывода многомерных функций

0 голосов
/ 24 октября 2018

Я хотел бы построить экстрактор объектов, используя CNN Caffe, и у меня уже есть большой выбор входных и желаемых выходных функций.

Теперь мне нужно обучить несколько сверточных слоев, чтобы научиться преобразовыватьвводить функции в вывод.

Мой вопрос: как я могу добиться этого в Caffe?

В качестве минимального примера, предположим, что я хотел обучить CNN, который инвертирует значения двумерного массива.

Например, если мой ввод 

[[0,1,0],
 [1,1,1],
 [0,1,0]]

, CNN должен выдавать 

[[1,0,1],
 [0,0,0],
 [1,0,1]].

Для 

[[0,0,0],
 [0,1,0],
 [0,0,0]]

вывод должен быть

[[1,1,1],
 [1,0,1],
 [1,1,1]]

и т. Д.

Конечно, это всего лишь минимальный пример, которым можно поделиться, настоящую проблему практически невозможно решить без использования нескольких сверток.

Я смог создать этот код для этой проблемы.В конце я использовал евклидову потерю, но, к сожалению, CNN ничего не изучает. 

ROOT_DIR = '/home'

from os.path import join
import numpy as np
import h5py
from itertools import product
import caffe
from caffe import layers
from caffe.proto import caffe_pb2

#%% GENERATE DATA
data_in = np.array([np.array(seq).reshape(1,3,3) for seq in product([0,1], repeat=9)])
data_out = np.array([-1*array+1 for array in data_in])

with open(join(ROOT_DIR, 'data.txt'), 'w') as ftxt:
    with h5py.File(join(ROOT_DIR, 'data.hdf5'), 'w') as fhdf5:
        fhdf5['data'] = data_in.astype(np.float32)
        fhdf5['label'] = data_out.astype(np.float32)
        ftxt.write(join(ROOT_DIR, 'data.hdf5'))

#%%DEFINE NET
net = caffe.NetSpec()
net.data, net.label = layers.HDF5Data(batch_size=64, source=join(ROOT_DIR, 'data.txt'), ntop=2)
net.conv1 = layers.Convolution(net.data, kernel_size=1, num_output=128)
net.relu1 = layers.ReLU(net.conv1, in_place=True)
net.conv2 = layers.Convolution(net.relu1, kernel_size=1, num_output=1)
net.relu2 = layers.ReLU(net.conv2, in_place=True)
net.loss = layers.EuclideanLoss(net.relu2, net.label)
net.to_proto()

with open(join(ROOT_DIR, 'invert_net.prototxt'), 'w') as f:
    f.write(str(net.to_proto()))

#%% DEFINE SOLVER
solver = caffe_pb2.SolverParameter()
solver.train_net = join(ROOT_DIR, 'invert_net.prototxt')
solver.max_iter = 10000
solver.base_lr = 0.01
solver.lr_policy = 'fixed'

with open(join(ROOT_DIR, 'solver.prototxt'), 'w') as f:
    f.write(str(solver))

#%% TRAIN NET
caffe.set_mode_cpu()
solver = caffe.SGDSolver(join(ROOT_DIR, 'solver.prototxt'))
solver.solve()
...