Guest User

Untitled

a guest
Apr 29th, 2020
127
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 1.92 KB | None | 0 0
  1. def build_generator_face(latent_dim=128*7*7, channels=3, face_sequence=True):
  2.  
  3.   model = Sequential()
  4.  
  5.   model.add(Dense(128 * 7 * 7, activation="relu", input_shape=(None, latent_dim)))
  6.   model.add(Reshape((7, 7, 128)))
  7.   model.add(UpSampling2D())
  8.   model.add(Conv2D(128, kernel_size=4, padding="same"))
  9.   model.add(BatchNormalization(momentum=0.8))
  10.   model.add(Activation("relu"))
  11.  
  12.   model.add(UpSampling2D())
  13.   model.add(Conv2D(64, kernel_size=4, padding="same"))
  14.   model.add(BatchNormalization(momentum=0.8))
  15.   model.add(Activation("relu"))
  16.  
  17.   if face_sequence == False:
  18.     model.add(Conv2D(64, kernel_size=4, padding="same"))
  19.     model.add(BatchNormalization(momentum=0.8))
  20.     model.add(Activation("relu"))
  21.  
  22.     model.add(Conv2D(64, kernel_size=4, padding="same"))
  23.     model.add(BatchNormalization(momentum=0.8))
  24.     model.add(Activation("relu"))
  25.  
  26.   else:
  27.     model.add(UpSampling2D(size=(1, 2)))
  28.     model.add(Conv2D(64, kernel_size=4, padding="same"))
  29.     model.add(BatchNormalization(momentum=0.8))
  30.     model.add(Activation("relu"))
  31.  
  32.     # go from 56 to 35 and continue upsampling
  33.  
  34.     model.add(Reshape((-1,3), input_shape=(28,56,3)))
  35.     model.add(Lambda(lambda x: x[:2940]))
  36.     model.add(Reshape((28,35,3)))
  37.  
  38.  
  39.     model.add(UpSampling2D(size=(1, 2)))
  40.     model.add(Conv2D(64, kernel_size=4, padding="same"))
  41.     model.add(BatchNormalization(momentum=0.8))
  42.     model.add(Activation("relu"))
  43.  
  44.     model.add(UpSampling2D(size=(1, 2)))
  45.     model.add(Conv2D(64, kernel_size=4, padding="same"))
  46.     model.add(BatchNormalization(momentum=0.8))
  47.     model.add(Activation("relu"))
  48.  
  49.     model.add(UpSampling2D(size=(1, 2)))
  50.     model.add(Conv2D(64, kernel_size=4, padding="same"))
  51.     model.add(BatchNormalization(momentum=0.8))
  52.     model.add(Activation("relu"))
  53.  
  54.  
  55.   model.add(Conv2D(channels, kernel_size=4, padding="same"))
  56.   model.add(Activation("tanh"))
  57.  
  58.   model.summary()
  59.  
  60.  
  61. build_generator_face()
Advertisement
Add Comment
Please, Sign In to add comment