Часть моих кодов:
output3d = Dense(5*2, name='imediate_2d', activation='tanh')(output2d)
def f(x):
# there are 10 numbers in input. 10 numbers are position information for 5 points.
# this function should return a 32*32 heatmap, where some positions are labelled as 1.
pass
proj_out = Lambda(lambda x:f(x), output_shape=(None, 32, 32), name='projection')(output2d)
Мне нужно закончить один пользовательский слой Keras.Ввод 5 (x, y) -координат для 5 точек.Мне нужно вернуть тепловую карту 32 * 32.Но я новичок в Керасе.Как заполнить вышеуказанную функцию f
.Любые уроки или советы приветствуются.Спасибо
ОБНОВЛЕНИЕ
Я провел поиск всю ночь, вход x
в слой Lambda tensor
, который я понятия не имею, как обрабатывать.Мне нужно что-то сложное, поэтому я не нашел никаких подсказок в документах keras / tenorflow.есть идеи как это сделать?Спасибо