Title:
IMAGE FEATURE EXTRACTION METHOD AND APPARATUS, AND ELECTRONIC DEVICE AND STORAGE MEDIUM
Document Type and Number:
WIPO Patent Application WO/2022/242127
Kind Code:
A1
Abstract:
Disclosed in the embodiments of the present invention are a method and apparatus for training an image feature extraction model, and a device and a storage medium. The method comprises: acquiring a plurality of groups of training sample data, wherein the training sample data comprises a sample extraction image and a sample feature image corresponding to the sample extraction image; and training a pre-established self-attention model on the basis of the plurality of groups of training sample data, so as to generate an image feature extraction model, wherein the self-attention model is used for learning a dependency relationship between each pixel point in the sample extraction image and all pixel points in the image. According to the technical solution, when an image feature extraction model is trained, a dependency relationship between each pixel point in a sample extraction image and all pixel points in the image can be effectively learned by means of a self-attention model, such that richer global context features of an image to be segmented are acquired, and the accuracy of training the image feature extraction model is improved.
Inventors:
LI YANG (CN)
WU JIANHUANG (CN)
WU JIANHUANG (CN)
Application Number:
PCT/CN2021/137818
Publication Date:
November 24, 2022
Filing Date:
December 14, 2021
Export Citation:
Assignee:
SHENZHEN INST OF ADV TECH CAS (CN)
International Classes:
G06K9/00
Foreign References:
| CN113326851A | 2021-08-31 | |||
| CN113159056A | 2021-07-23 | |||
| CN111667495A | 2020-09-15 | |||
| KR102278756B1 | 2021-07-16 |
Other References:
WANG, XIAOLONG ET AL.: "Non-local Neural Networks", CVPR 2018, 22 June 2018 (2018-06-22), pages 7794 - 7803, XP093000642, DOI: 10.1109/CVPR.2018.00813
ANONYMOUS: "CV AttentionNon-Local neural networks", UNDERSTANDING AND IMPLEMENTATION OF NON-LOCAL NEURAL NETWORKS, 5 January 2020 (2020-01-05), pages 1 - 5, XP093009214, Retrieved from the Internet [retrieved on 20220210]
ANONYMOUS: "CV AttentionNon-Local neural networks", UNDERSTANDING AND IMPLEMENTATION OF NON-LOCAL NEURAL NETWORKS, 5 January 2020 (2020-01-05), pages 1 - 5, XP093009214, Retrieved from the Internet
Attorney, Agent or Firm:
BEIJING ZHONG XUN TONG DA INTELLECTUAL PROPERTY AGENCY CO., LTD. (CN)
Download PDF: