Method and apparatus for training image segmentation model, computer device, and storage medium
US11961233B2 · kind B2 · utility
Assignee
Inventor
Key dates
| Filing date | Sep 9, 2021 |
| Grant date | Apr 16, 2024 |
| Priority date | — |
| Expiry date | Aug 17, 2042 |
Classification
- Technology area (CPC G)Physics
- CPC primaryG06V2201/031
- WIPO fieldComputer technology
- WIPO sectorElectrical engineering
Abstract
This application provides a method and apparatus for training an image segmentation model, a device, and a storage medium. The method includes: training an initial image segmentation model by using source domain samples, to obtain a pre-trained image segmentation model; extracting a predicted segmentation result of a source domain image and a predicted segmentation result of a target domain image by using the pre-trained image segmentation model; training a first discriminator by using the predicted segmentation result of the source domain image and the predicted segmentation result of the target domain image; training a second discriminator by using the predicted segmentation result of the source domain image and a standard segmentation result of the source domain image; and iteratively training the pre-trained image segmentation model according to a loss function of the pre-trained image segmentation model, an adversarial loss function of the first discriminator, and an adversarial loss function of the second discriminator, until convergence, to obtain a trained image segmentation model.
Source: USPTO / EPO open patent data. Objective bibliographic and citation counts.