Eureka AIR delivers breakthrough ideas for toughest innovation challenges, trusted by R&D personnel around the world.

Model training method and device, target detection method and device, equipment and storage medium

A training method and model technology, applied in the field of image processing, can solve the problems of difficult detection and classification, poor model generalization ability, low model adaptability, etc., to reduce training costs, reduce labeling workload, and improve generalization ability. Effect

Pending Publication Date: 2021-04-30
MEGVII BEIJINGTECH CO LTD
View PDF0 Cites 1 Cited by
  • Summary
  • Abstract
  • Description
  • Claims
  • Application Information

AI Technical Summary

Problems solved by technology

[0004] However, when the preset model is trained in the above way, the generalization ability of the final model is poor
For example, technicians select a large number of pictures of country A's road field as sample pictures to train the target detection model, and the obtained expected model has better detection and classification capabilities for each object in the picture of country A's road field, but for It is difficult to achieve accurate detection and classification of various objects in the pictures of the road field in country B, and the adaptability of the model is low

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
  • Model training method and device, target detection method and device, equipment and storage medium
  • Model training method and device, target detection method and device, equipment and storage medium
  • Model training method and device, target detection method and device, equipment and storage medium

Examples

Experimental program
Comparison scheme
Effect test

Embodiment Construction

[0058] The following will clearly and completely describe the technical solutions in the embodiments of the present application with reference to the drawings in the embodiments of the present application. Obviously, the described embodiments are part of the embodiments of the present application, not all of them. Based on the embodiments in this application, all other embodiments obtained by persons of ordinary skill in the art without creative efforts fall within the protection scope of this application.

[0059] In the field of machine learning, technicians collect a large number of sample pictures, mark the sample pictures, and train the preset model through the marked sample pictures, so as to obtain the target model with the desired function. images for detection, classification, prediction, and more. During this period, the quality of model training usually depends on the structure and type of the preset model built by the technician, and also depends on the way the tec...

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 embodiment of the invention relates to a detection model training method and device, equipment and a storage medium, and aims to improve the generalization ability of a model. The method comprises the steps of obtaining a plurality of source domain sample images and a plurality of target domain sample images, wherein each source domain sample image comprises annotation box information of a pre-marked source domain object; inputting the source domain sample image and the target domain sample image into a feature extraction model in pairs to obtain a first feature map and a first detection frame of the source domain sample image and a second feature map and a second detection frame of the target domain sample image; judging the domain category of each pixel point in each feature map to obtain a plurality of first judgment results; judging the domain category of each detection box to obtain a plurality of second judgment results; and updating the feature extraction model according to the first detection box and annotation box information and according to the plurality of first judgment results and the plurality of second judgment results.

Description

technical field [0001] The embodiments of the present application relate to the technical field of image processing, and in particular, relate to a model training method, a target detection method, a device, a device, and a storage medium. Background technique [0002] With the development of image processing technology and the improvement of the computing power of hardware equipment, in order to enable the machine to intelligently realize functions such as detection, classification, and prediction, technicians collect a large number of sample pictures and mark the sample pictures. The sample images are used to train the built preset model to obtain a model with desired functions. During this period, the quality of model training usually depends on the structure and type of the preset model built by the technician, and also depends on the way the technician selects and marks the sample pictures. [0003] Taking target detection based on road scenes as an example, technician...

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
IPC IPC(8): G06K9/46G06K9/62G06N3/04G06N3/08
CPCG06N3/04G06N3/08G06V10/44G06F18/241Y02T10/40
Inventor 李昂杨学张志强俞刚
Owner MEGVII BEIJINGTECH CO LTD
Who we serve
  • R&D Engineer
  • R&D Manager
  • IP Professional
Why Eureka
  • Industry Leading Data Capabilities
  • Powerful AI technology
  • Patent DNA Extraction
Social media
Eureka Blog
Learn More
PatSnap group products