Skip to content

Commit

Permalink
Fix typo and wrong reference to UNet
Browse files Browse the repository at this point in the history
  • Loading branch information
mr-ubik committed Jan 28, 2020
1 parent f8865b6 commit 1bf4103
Show file tree
Hide file tree
Showing 2 changed files with 9 additions and 4 deletions.
2 changes: 1 addition & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -457,7 +457,7 @@ tuple(tuple(a,b), noise)

Where `a` is the input sample, `b` is the label/condition (if any, otherwise fill it with `0`), and `noise` is the latent vector of input.

To train Pix2Pix-like architecture, that have no `noise` as ConvGenerator input, just return the values in thee format `(tuple(a,b), b)` since the condition is the generator output.
To train Pix2Pix-like architecture, that have no `noise` as ConvGenerator input, just return the values in thee format `(tuple(a,b), b)` since the condition is the generator input.

## Test
In order to run the tests (with the doctests), linting and docs generation simply use `tox`.
Expand Down
11 changes: 8 additions & 3 deletions src/ashpy/models/convolutional/unet.py
Original file line number Diff line number Diff line change
Expand Up @@ -27,7 +27,10 @@ class UNet(Conv2DInterface):
"""
UNet Architecture.
Used in Image-to-Image Translation with Conditional Adversarial Nets [1]_.
Architecture similar to the one found in "Image-to-Image Translation
with Conditional Adversarial Nets" [1]_.
Originally proposed in "U-Net: Convolutional Networks for Biomedical Image Segmentation" [2]_.
Examples:
* Direct Usage:
Expand All @@ -50,8 +53,10 @@ class UNet(Conv2DInterface):
(1, 512, 512, 3)
True
.. [1] Image-to-Image Translation with Conditional Adversarial Nets
https://arxiv.org/abs/1611.04076
.. [1] Image-to-Image Translation with Conditional Adversarial Nets -
https://arxiv.org/abs/1611.07004
.. [2] U-Net: Convolutional Networks for Biomedical Image Segmentation -
https://arxiv.org/abs/1505.04597
"""

Expand Down

0 comments on commit 1bf4103

Please sign in to comment.