from keras.models import Sequential
from keras.models import Model
from keras.layers import Input, Convolution3D

import  keras
import h5py
from keras.optimizers import Adam

def srcnn(input_shape=(33,33,110,1)):
    #for ROSIS  sensor
    model = Sequential()
    model.add(Convolution3D(64, 9, 9, 7, input_shape=input_shape, activation='relu'))
    model.add(Convolution3D(32, 1, 1, 1, activation='relu'))
    model.add(Convolution3D(9, 1, 1, 1, activation='relu'))
    model.add(Convolution3D(1, 5, 5, 3))
    model.compile(Adam(lr=0.00005), 'mse')
    return model