Supercharge Your Innovation With Domain-Expert AI Agents!

A semantic segmentation model training method, computer equipment and storage medium

A computer equipment and semantic segmentation technology, applied in the field of image processing, can solve problems such as network training instability, and achieve the effect of avoiding gradient instability and preventing category imbalance.

Active Publication Date: 2021-08-31
XIAMEN UNIV
View PDF9 Cites 0 Cited by
  • Summary
  • Abstract
  • Description
  • Claims
  • Application Information

AI Technical Summary

Problems solved by technology

[0010] The purpose of the present invention is to provide a semantic segmentation model training method, computer equipment and storage media to solve the problem of network training instability caused by using Dice_loss in the prior art

Method used

the structure of the environmentally friendly knitted fabric provided by the present invention; figure 2 Flow chart of the yarn wrapping machine for environmentally friendly knitted fabrics and storage devices; image 3 Is the parameter map of the yarn covering machine
View more

Image

Smart Image Click on the blue labels to locate them in the text.
Viewing Examples
Smart Image
  • A semantic segmentation model training method, computer equipment and storage medium
  • A semantic segmentation model training method, computer equipment and storage medium
  • A semantic segmentation model training method, computer equipment and storage medium

Examples

Experimental program
Comparison scheme
Effect test

Embodiment

[0026] Such as figure 1 Shown is the model training flowchart for semantic segmentation. The parameters of the model can be used to calculate the degree of deviation between the predicted value and the target value, and a loss function can be formed based on these parameters. In the present invention, the loss function is defined and optimized, the model parameters are updated, and the model is finally converged in the training segmentation model (ie, S2).

[0027] Define R n × R n → Functions on R Among them, α is called the adjustment factor, and its value range is between It can be proved (R n ,D) form the distance space and are used in the backpropagation algorithm.

[0028] First, prove that the function D(P,T) is R n distance on:

[0029] obviously, yes T∈R n , there exists D(P,T)=D(T,P), D(P,T)≥0, and the equality sign is obtained only when P=T.

[0030] Next, prove that Z, T ∈ R n , all exist D(P,T)≤D(P,Z)+D(Z,T).

[0031] Note that the latter part of...

the structure of the environmentally friendly knitted fabric provided by the present invention; figure 2 Flow chart of the yarn wrapping machine for environmentally friendly knitted fabrics and storage devices; image 3 Is the parameter map of the yarn covering machine
Login to View More

PUM

No PUM Login to View More

Abstract

The invention discloses a semantic segmentation model training method, the loss function of which is as follows: in the formula, P is a prediction vector, T is a one-hot coded label vector; a and b are weights, and a increases with the increase of the training period Large, b decreases with the increase of the training cycle, and a+b=1; α is an adjustment factor, and its value range is in the present invention. A function is defined, and it is combined with the Dice coefficient to obtain a new loss function, it is not easy to have the problem of gradient explosion or disappearance in the process of backpropagation, and at the same time, it is sensitive to whether the classification of a single pixel point is correct or not, and is sensitive to whether the overall classification is correct or not, avoiding the gradient that occurs when using the Dice coefficient alone as a loss function Unstable, and can prevent category imbalance.

Description

technical field [0001] The invention relates to the field of image processing, in particular to a semantic segmentation model training method, computer equipment and storage media. Background technique [0002] In deep learning semantic segmentation tasks, the Dice coefficient is a commonly used measurement method for evaluating segmentation results. Its derived loss function Dice_loss directly aims to optimize the intersection ratio, which is not easily affected by analog imbalance, and is widely used in network training. However, Dice_loss sometimes has a sharp gradient change during network training, which leads to unstable network training. Therefore, this application analyzes the reasons for its occurrence and proposes an improvement plan. [0003] For the binary classification task, record the prediction vector as P, and the one-hot encoded label vector as T. Dice_loss can be expressed as follows: [0004] [0005] In the process of backpropagation, the loss funct...

Claims

the structure of the environmentally friendly knitted fabric provided by the present invention; figure 2 Flow chart of the yarn wrapping machine for environmentally friendly knitted fabrics and storage devices; image 3 Is the parameter map of the yarn covering machine
Login to View More

Application Information

Patent Timeline
no application Login to View More
Patent Type & Authority Patents(China)
IPC IPC(8): G06K9/62G06T7/11
CPCG06T7/11G06F18/214
Inventor 周联昱
Owner XIAMEN UNIV
Features
  • R&D
  • Intellectual Property
  • Life Sciences
  • Materials
  • Tech Scout
Why Patsnap Eureka
  • Unparalleled Data Quality
  • Higher Quality Content
  • 60% Fewer Hallucinations
Social media
Patsnap Eureka Blog
Learn More

Browse by: Latest US Patents, China's latest patents, Technical Efficacy Thesaurus, Application Domain, Technology Topic, Popular Technical Reports.

© 2025 PatSnap. All rights reserved.Legal|Privacy policy|Modern Slavery Act Transparency Statement|Sitemap|About US| Contact US: help@patsnap.com