Training method and device suitable for industrial part recognition model and storage medium

Abstract

The invention provides a training method and device suitable for an industrial part recognition model and a storage medium. The training method comprises the steps that a preset industrial part three-dimensional image in an industrial part database is acquired; performing two-dimensional processing on the three-dimensional image of the preset industrial part to generate a plurality of two-dimensional images of the preset industrial part, wherein each two-dimensional image of the preset industrial part is an image of the same industrial part at different angles; and training the industrial part recognition model by taking the preset industrial part two-dimensional images of all the same industrial part at different angles as training samples. Obtaining a customized industrial part three-dimensional image customized by a user at the current moment; performing two-dimensional processing on the three-dimensional image of the customized industrial part to generate a plurality of two-dimensional images of the customized industrial part, wherein each two-dimensional image of the customized industrial part is an image of the same industrial part at different angles; and training the industrial part recognition model again by taking the customized industrial part two-dimensional images of all the same industrial part at different angles as training samples.

Description

technical field [0001] The invention relates to model training technology, in particular to a training method, device and storage medium suitable for industrial component recognition models. Background technique [0002] Industrial components are a very important pillar industry in my country's national economy, with an output value of over one trillion yuan. Among them, fasteners (commonly known as screws) are the most widely used and involve industries, ranging from mobile phones to airplanes. Fasteners are required for connection. As an industrial basic supporting industry, fasteners are also dubbed the title of "industrial rice". Each country, region and industry has its own standards for fastener products (standard content includes information such as technology, performance, application, drawings, etc.), my country is a big industrial manufacturing country, which gathers global fastener product data. There are more than 8,000 fastener standards in the world, and the comm...

AI Technical Summary

Problems solved by technology

However, after the database is formed, the applicability of the model for parts identification and query based on the database is poor, and the corresponding industrial parts and their corresponding information cannot be found in the database, and the recognition accuracy is poor.

Images