提交 564b8c6c 编写于 作者: Z zchen0211

gan api

上级 d0f719f7
...@@ -19,12 +19,12 @@ In our GAN design, we wrap it as a user-friendly easily customized python API to ...@@ -19,12 +19,12 @@ In our GAN design, we wrap it as a user-friendly easily customized python API to
| softmax loss (done) | ? | Y | | softmax loss (done) | ? | Y |
| reshape op (done) | ? | Y | | reshape op (done) | ? | Y |
| Dependency Engine (done) | Jiayi | Y * | | Dependency Engine (done) | Jiayi | Y * |
| Python API (done) | Jiayi | Y * | | Python API (done) | Longfei, Jiayi | Y * |
| Executor (done) | Tony | Y * | | Executor (done) | Tony | Y * |
| Multi optimizer | ? | Y * | | Multi optimizer (woking) | Longfei | Y * |
| Optimizer with any para | ? | Y * | | Optimizer with any para | ? | Y * |
| Concat op | ? | N (Cond) | | Concat op (done) | ? | N (Cond) |
| Repmat op | ? | N (Cond) | | Repmat op (done) | ? | N (Cond) |
<p align="center"> <p align="center">
...@@ -91,7 +91,8 @@ class DCGAN(object): ...@@ -91,7 +91,8 @@ class DCGAN(object):
- Concatenation, batch-norm, FC operations required; - Concatenation, batch-norm, FC operations required;
- Deconv layer required, which is missing now... - Deconv layer required, which is missing now...
```python ```python
def generator(self, z, y = None): class DCGAN(object):
def generator(self, z, y = None):
# input z: the random noise # input z: the random noise
# input y: input data label (optional) # input y: input data label (optional)
# output G_im: generated fake images # output G_im: generated fake images
...@@ -116,7 +117,8 @@ def generator(self, z, y = None): ...@@ -116,7 +117,8 @@ def generator(self, z, y = None):
- Given a noisy input z, returns a fake image. - Given a noisy input z, returns a fake image.
- Concatenation, Convolution, batch-norm, FC, Leaky-ReLU operations required; - Concatenation, Convolution, batch-norm, FC, Leaky-ReLU operations required;
```python ```python
def discriminator(self, image): class DCGAN(object):
def discriminator(self, image):
# input image: either generated images or real ones # input image: either generated images or real ones
# output D_h2: binary logit of the label # output D_h2: binary logit of the label
...@@ -137,8 +139,8 @@ def discriminator(self, image): ...@@ -137,8 +139,8 @@ def discriminator(self, image):
- Build generator and discriminators; - Build generator and discriminators;
- Define two training losses for discriminator and generator, respectively. - Define two training losses for discriminator and generator, respectively.
```python ```python
def build_model(self): class DCGAN(object):
def build_model(self):
# input data # input data
if self.y_dim: if self.y_dim:
self.y = pd.data(pd.float32, [self.batch_size, self.y_dim]) self.y = pd.data(pd.float32, [self.batch_size, self.y_dim])
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册