Lightweight target detection method and device based on task interaction
By introducing a lightweight task interaction structure into the object detection model, the number of model parameters and calculation complexity are reduced, and the problems of large number of model parameters and high calculation amount in the prior art are solved, efficient object detection is achieved and hardware requirements and training costs are reduced.
Patent Information
- Application Number
- CN202411587471.7
- Authority / Receiving Office
- CN · China
- Patent Type
- Applications(China)
- Current Assignee / Owner
- Filing Date
- 2024-11-08
- Publication Date
- 2025-03-28
- Estimated Expiration
- 2044-11-08
AI Technical Summary
The existing open source object detection model has large model parameters and high computational volume due to the output of the feature pyramid or the use of upper-level features of the classification branch/positioning branch, which increases hardware requirements and actual application costs.
A lightweight object detection method based on task interaction is proposed. Through the model composed of backbone network, neck network, task interaction lightweight structure and detection head, the number of model parameters is reduced and the detection accuracy is maintained. The specific steps include acquiring image data, performing feature extraction, fusion and task interaction processing, generating lightweight task feature data, and inputting a detection head for prediction.
While reducing the amount of model parameters, it maintains a high target detection accuracy, reduces the computational complexity and memory usage, improves the model's inference speed, reduces hardware requirements, and reduces the model's training cost.