#--------------------------------------------#
#   该部分代码只用于看网络结构，并非测试代码
#--------------------------------------------#
from nets.unet import Unet

if __name__ == "__main__":
    model = Unet([512, 512, 3], 21)
    model.summary()

    # for i,layer in enumerate(model.layers):
    #     print(i,layer.name)
