Target detection model training method and device and electronic equipment

A technology of target detection and training method, which is applied in the information field and can solve the problems of poor detection accuracy of target detection models.

Pending Publication Date: 2020-02-21
BEIJING DAJIA INTERNET INFORMATION TECH CO LTD
View PDF4 Cites 17 Cited by
  • Summary
  • Abstract
  • Description
  • Claims
  • Application Information

AI Technical Summary

Problems solved by technology

[0004] The purpose of the embodiments of the present application is to provide a training method, device and electronic e

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
  • Target detection model training method and device and electronic equipment
  • Target detection model training method and device and electronic equipment
  • Target detection model training method and device and electronic equipment

Examples

Experimental program
Comparison scheme
Effect test

Embodiment Construction

[0071] In order to enable ordinary persons in the art to better understand the technical solutions of the present disclosure, the technical solutions in the embodiments of the present disclosure will be clearly and completely described below in conjunction with the accompanying drawings.

[0072] It should be noted that the terms "first" and "second" in the specification and claims of the present disclosure and the above drawings are used to distinguish similar objects, but not necessarily used to describe a specific sequence or sequence. It is to be understood that the data so used are interchangeable under appropriate circumstances such that the embodiments of the disclosure described herein can be practiced in sequences other than those illustrated or described herein. The implementations described in the following exemplary examples do not represent all implementations consistent with the present disclosure. Rather, they are merely examples of apparatuses and methods consi...

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 provides a target detection model training method and device and electronic equipment, and the method comprises the steps: selecting a sample image from a preset training image set, inputting the sample image into a to-be-trained network model, and obtaining the prediction position information and prediction probability of a prediction target in the sample image; calculating a first loss of the to-be-trained network model according to the prediction position information and a preset regression function; calculating a second loss of the to-be-trained network model according to the prediction probability and a preset focus loss function; calculating target loss of the to-be-trained network model through a preset target loss function; and adjusting the to-be-trained network model according to the target loss to obtain a trained network model. According to the technical scheme of the invention, the weight correction is carried out according to the proportions of different positive samples through the preset focus loss function, so that the model obtained through training can adapt to different positive sample proportions, and the detection precision of the target in the to-be-detected image is improved.

Description

technical field [0001] The present application relates to the field of information technology, in particular to a method, device and electronic equipment for training a target detection model. Background technique [0002] Currently, object detection, as a key technology in the field of computer vision, is widely used in many applications. Especially with the acceleration of mobile networks and the popularization of mobile devices in recent years, there are more and more demands for running object detection algorithms on mobile devices and other devices. Targets such as people or objects in the image can be detected through the target detection algorithm. Generally, multiple target candidate areas are obtained to judge whether each candidate area is a target or a mixed loss function of classification and regression is used for target detection. [0003] However, the method of judging whether each candidate area is a target by obtaining multiple target candidate areas requir...

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): G06T7/00
CPCG06T7/0002G06T2207/10004G06T2207/20081G06T2207/30168
Inventor 冯文雅宋丛礼
Owner BEIJING DAJIA INTERNET INFORMATION TECH 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
Try Eureka
PatSnap group products