from src.model import unpool_resize,unpool_deconv, unpool_checkerboard, unpool_simple
from tensorflow.keras.layers import Input, UpSampling2D
from tensorflow.keras.models import Model

input = Input(shape=(20, 20, 3))

out1 = unpool_resize(input)
model1 = Model(inputs=input, outputs=out1)
print("")

out2 = unpool_deconv(input,512)
model2 = Model(inputs=input, outputs=out2)
print("")

out3 = UpSampling2D((2,2))(input)
out3 = unpool_checkerboard(out3)
model3 = Model(inputs=input, outputs=out3)
print("")


tensorflow.keras.layers.UpSampling2D tensorflow.keras.models.Model tensorflow.keras.layers.Input
[(5, 'tensorflow.keras.layers.Input', 'Input', ([], {'shape': '(20, 20, 3)'}), False, 'from tensorflow.keras.layers import Input, UpSampling2D\n'), (7, 'src.model.unpool_resize', 'unpool_resize', (['input'], {}), False, 'from src.model import unpool_resize, unpool_deconv, unpool_checkerboard, unpool_simple\n'), (8, 'tensorflow.keras.models.Model', 'Model', ([], {'inputs': 'input', 'outputs': 'out1'}), False, 'from tensorflow.keras.models import Model\n'), (11, 'src.model.unpool_deconv', 'unpool_deconv', (['input', '(512)'], {}), False, 'from src.model import unpool_resize, unpool_deconv, unpool_checkerboard, unpool_simple\n'), (12, 'tensorflow.keras.models.Model', 'Model', ([], {'inputs': 'input', 'outputs': 'out2'}), False, 'from tensorflow.keras.models import Model\n'), (16, 'src.model.unpool_checkerboard', 'unpool_checkerboard', (['out3'], {}), False, 'from src.model import unpool_resize, unpool_deconv, unpool_checkerboard, unpool_simple\n'), (17, 'tensorflow.keras.models.Model', 'Model', ([], {'inputs': 'input', 'outputs': 'out3'}), False, 'from tensorflow.keras.models import Model\n'), (15, 'tensorflow.keras.layers.UpSampling2D', 'UpSampling2D', (['(2, 2)'], {}), False, 'from tensorflow.keras.layers import Input, UpSampling2D\n')]