Skip to content

Commit a5acd55

Browse files
committed
Update docs + Colab inference
1 parent b42df9d commit a5acd55

File tree

2 files changed

+207
-180
lines changed

2 files changed

+207
-180
lines changed

docs/source/guides/training_wnet.rst

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -20,21 +20,21 @@ You may find below some guidelines, based on our own data and testing.
2020

2121
The WNet3D is designed to segment objects based on their brightness, and is particularly well-suited for images with a clear contrast between objects and background.
2222

23-
The WNet3D is not suitable for images with artifacts, therefore care should be taken that the images are clean and that the objects are at least somewhat distinguishable from the background.
23+
The WNet3D is not suitable for images with strong noise or artifacts, therefore care should be taken that the images are clean and that the objects are at least somewhat distinguishable from the background.
2424

2525

2626
.. important::
2727
For optimal performance, the following should be avoided for training:
2828

2929
- Images with very large, bright regions
30-
- Almost-empty and empty images
30+
- Almost-empty and empty images, especially if noise is present
3131
- Images with large empty regions or "holes"
3232

33-
However, the model may be accomodate:
33+
However, the model may accomodate:
3434

3535
- Uneven brightness distribution
3636
- Varied object shapes and radius
37-
- Noisy images
37+
- Noisy images (as long as resolution is sufficient and boundaries are clear)
3838
- Uneven illumination across the image
3939

4040
For optimal results, during inference, images should be similar to those the model was trained on; however this is not a strict requirement.
@@ -88,7 +88,7 @@ Common issues troubleshooting
8888
If you do not find a satisfactory answer here, please do not hesitate to `open an issue`_ on GitHub.
8989

9090

91-
- **The NCuts loss "explodes" after a few epochs** : Lower the learning rate, for example start with a factor of two, then ten.
91+
- **The NCuts loss "explodes" upward after a few epochs** : Lower the learning rate, for example start with a factor of two, then ten.
9292

9393
- **Reconstruction (decoder) performance is poor** : First, try increasing the weight of the reconstruction loss. If this is ineffective, switch to BCE loss and set the scaling factor of the reconstruction loss to 0.5, OR adjust the weight of the MSE loss.
9494

0 commit comments

Comments
 (0)