Official websites use .gov
A
.gov website belongs to an official
government organization in the United States.
Secure .gov websites use HTTPS
A lock (
) or https:// means you've safely
connected to the .gov website. Share sensitive
information only on official, secure websites.
Main parameters of U-Net training model.
| Main Parameter | Values |
|---|---|
| Num classes | 2 |
| Input shape | (512, 512) |
| Epoch | 50 |
| Batch size | 5 |
| Initial learning rate | 0.0001 |
| Optimizer type | Adam |
| momentum | 0.9 |
| Save period | 5 |
| Number of workers | 4 |
| Number of train sets | 3264 |
| Number of validation sets | 816 |
