Multi-feature-based granite rock identification method, system, device and equipment

Abstract

The invention discloses a granite rock recognition method, system, device and equipment based on multiple features, relates to the technical field of image recognition, and can solve the technical problems that granite rock recognition depends on experience and knowledge of professionals, time is consumed and efficiency is low. Comprising the following steps: acquiring a to-be-trained rock image, and performing multi-level feature label labeling on the to-be-trained rock image to obtain a labeled rock image; constructing an initial model for correspondingly identifying the features of each level, and training and testing the initial model by using the marked rock image to obtain a target model for correspondingly identifying the features of each level; and obtaining to-be-identified rock images, and identifying the to-be-identified rock images in sequence according to the target models corresponding to the feature level sequence to obtain target multilevel features of the to-be-identified rock images.

Description

Technical Field

[0001] The present invention relates to the technical field of image recognition, and in particular to a method, system, device and equipment for identifying granite rocks based on multiple features. Background Art

[0002] Granite is a general term for granite, which refers to granite and its closely associated mineral components, which are mainly quartz (>5%) and feldspar. Common granite rocks include the following: alkali feldspar granite, potassium feldspar granite, monzonitic granite, granodiorite, and plagioclase granite.

[0003] Lithology identification is a routine task in geological work. Traditionally, this work is mainly completed by geological experts through comprehensive identification of rock color, structure, texture and grain size characteristics based on experience. The main drawback of this method is that it is time-consuming and labor-intensive, and has high requirements for the professional knowledge of geologists. Summary of the invention

[0004] In view of this, the present invention provides a method, system, device and equipment for identifying granite rocks based on multiple features, which can solve the technical problem that the identification of granite rocks relies on the experience and knowledge of professionals, is time-consuming and inefficient.

[0005] According to a first aspect of the present invention, a method for identifying granite rocks based on multiple features is provided, the method comprising:

[0006] Acquire a rock image to be trained, and perform multi-level feature labeling on the rock image to be trained to obtain a labeled rock image;

[0007] Constructing an initial model corresponding to each level of feature recognition, and using the annotated rock images to train and test the initial model respectively, to obtain a target model corresponding to each level of feature recognition;

[0008] The rock image to be identified is acquired, and the rock image to be identified is identified in sequence according to the target model corresponding to the feature level sequence to obtain the target multi-level features of the rock image to be identified.

[0009] Preferably, the step of labeling the to-be-trained rock image with multi-level features to obtain a labeled rock image includes:

[0010] Determine a multi-level feature, wherein the multi-level feature includes: rock and non-rock as a first-level feature, igneous rock and non-igneous rock as a second-level feature, granite and non-granite as a third-level feature, and lithology of granite as a fourth-level feature;

[0011] The rock image to be trained is labeled with features step by step to obtain a labeled rock image.

[0012] Preferably, the constructing of an initial model corresponding to identifying each level of features includes:

[0013] constructing a first initial model for identifying rocks and non-rocks;

[0014] constructing a second initial model for identifying igneous rocks and non-igneous rocks;

[0015] Constructing a third initial model for identifying granitoids and non-granitoids;

[0016] constructing a fourth initial model for identifying lithology of granitoids;

[0017] The first initial model, the second initial model, the third initial model and the fourth initial model are used as initial models corresponding to each level of feature identification.

[0018] Preferably, the initial model is trained and tested using the annotated rock image to obtain a target model corresponding to each level of feature recognition, including:

[0019] Using the annotated rock image to train and test the first initial model, to obtain a first target model corresponding to identifying the first level features;

[0020] Deleting the image whose first-level feature is non-rock in the labeled rock image to obtain a first deleted image;

[0021] Using the first deleted image to train and test the second initial model, to obtain a second target model corresponding to identifying the second level characteristics;

[0022] Deleting the image whose second-level feature is non-magmatic rock in the first deleted image to obtain a second deleted image;

[0023] Using the second deleted image to train and test the third initial model, to obtain a third target model corresponding to the recognition of the third level characteristics;

[0024] Deleting the image whose third-level feature is not granite in the second deleted image to obtain a third deleted image;

[0025] Using the third deleted image to train and test the fourth initial model, to obtain a fourth target model corresponding to the fourth level feature recognition;

[0026] The first target model, the second target model, the third target model and the fourth target model are used as target models corresponding to each level of features to be identified.

[0027] Preferably, the target model corresponding to the feature level sequence sequentially identifies the rock image to be identified to obtain the target multi-level features of the rock image to be identified, including:

[0028] Identify the rock image to be identified according to the first target model to obtain an image of a target rock and an image of a target non-rock in the rock image to be identified;

[0029] Recognize the image of the target rock according to the second target model to obtain an image of a target igneous rock and an image of a target non-igneous rock in the image of the target rock;

[0030] Recognize the image of the target igneous rock according to the third target model to obtain a target granite image and a target non-granite image in the image of the target igneous rock;

[0031] Recognize the target granite image according to the fourth target model to obtain the target lithology of the target granite image;

[0032] The target multi-level features of the rock image to be identified are obtained according to the target rock image, the target non-rock image, the target igneous rock image, the target non-igneous rock image, the target granite image, the target non-granite image and the target lithology.

[0033] According to a second aspect of the present invention, a granite rock identification system based on multiple features is provided, the system comprising:

[0034] Rock image management module, recognition module and model management module to be trained;

[0035] The to-be-trained rock image management module is used to respond to a user's request for a management operation on the to-be-trained rock image, and perform a corresponding management operation on the to-be-trained rock image, wherein the management operation request includes setting the number and type of the to-be-trained rock images, and adding, deleting, modifying, and querying the to-be-trained rock images;

[0036] The recognition module is used to receive the rock image to be recognized, and recognize the rock image to be recognized in sequence according to the target model corresponding to the feature level sequence, so as to obtain the target multi-level features of the rock image to be recognized;

[0037] The model management module is used to re-train the target model when the number of rock images to be trained newly added or modified by the user is greater than a preset threshold, and during training, receive hyperparameters or automatically search for the hyperparameters to obtain an updated target model, so that the recognition model receives the rock images to be recognized, and recognizes the rock images to be recognized in turn according to the updated target models corresponding to the feature level sequence, to obtain updated target multi-level features of the rock images to be recognized.

[0038] According to a third aspect of the present invention, a granite rock identification device based on multiple features is provided, the device comprising:

[0039] An acquisition module is used to acquire a rock image, and label the rock image with multi-level features to obtain a labeled rock image;

[0040] A training module is used to construct an initial model corresponding to each level of feature recognition, and to train and test the initial model using the annotated rock image to obtain a target model corresponding to each level of feature recognition;

[0041] The application module is used to obtain the rock image to be identified, identify the rock image to be identified in turn according to the target model corresponding to the feature level sequence, and obtain the target multi-level features of the rock image to be identified.

[0042] Preferably, the acquisition module includes a determination unit and a marking unit;

[0043] The determination unit is used to determine a multi-level feature, wherein the multi-level feature includes: rock and non-rock as a first-level feature, igneous rock and non-igneous rock as a second-level feature, granite and non-granite as a third-level feature, and the lithology of granite as a fourth-level feature;

[0044] The labeling unit is used to label the features of the rock image to be trained step by step to obtain a labeled rock image.

[0045] According to a fourth aspect of the present application, a storage medium is provided, on which a computer program is stored. When the computer program is executed by a processor, the above-mentioned multi-feature-based granite rock identification method is implemented.

[0046] According to the fifth aspect of the present application, a computer device is provided, including a storage medium, a processor, and a computer program stored on the storage medium and executable on the processor, wherein the processor implements the above-mentioned multi-feature-based granite rock identification method when executing the program.

[0047] By means of the above technical scheme, the present invention provides a method, system, device and equipment for granite rock recognition based on multiple features. First, a rock image to be trained is obtained, and the rock image to be trained is labeled with multi-level features to obtain a labeled rock image; then an initial model corresponding to each level of feature recognition is constructed, and the initial model is trained and tested using the labeled rock image to obtain a target model corresponding to each level of feature recognition; finally, a rock image to be recognized is obtained, and the rock image to be recognized is sequentially recognized according to the target model corresponding to the feature level sequence to obtain the target multi-level features of the rock image to be recognized. By means of the technical scheme of the present invention, the rock image to be trained is labeled with multi-level features, and the geological characteristics of the rock itself are fully considered. For each level of feature, an initial model is trained to obtain a target model, and the target model is used to specifically recognize the features of the rock image to be recognized, thereby improving the recognition accuracy and improving the recognition efficiency compared to manual recognition.

[0048] The above description is only an overview of the technical solution of the present invention. In order to more clearly understand the technical means of the present invention, it can be implemented according to the contents of the specification. In order to make the above and other purposes, features and advantages of the present invention more obvious and easy to understand, the specific implementation methods of the present invention are listed below. BRIEF DESCRIPTION OF THE DRAWINGS

[0049] The drawings described herein are used to provide a further understanding of the present invention and constitute a part of the present invention. The exemplary embodiments of the present invention and their descriptions are used to explain the present invention and do not constitute an improper limitation on the local application. In the drawings:

[0050] Figure 1 A schematic diagram of a process of a granite rock identification method based on multiple features provided by an embodiment of the present invention is shown;

[0051] Figure 2 A schematic flow chart of another granite rock identification method based on multiple features provided by an embodiment of the present invention is shown;

[0052] Figure 3 A schematic structural diagram of a granite rock identification device based on multiple features provided by an embodiment of the present invention is shown;

[0053] Figure 4 A schematic structural diagram of another granite rock identification device based on multiple features provided by an embodiment of the present invention is shown;

[0054] Figure 5 A schematic diagram of a process for identifying granite lithology by using a target model provided by an embodiment of the present invention is shown;

[0055] Figure 6 The following is a diagram showing the architecture of a granite rock identification system based on multiple features provided by an embodiment of the present invention;

[0056] Figure 7 A functional diagram of a granite rock identification system based on multiple features provided by an embodiment of the present invention is shown;

[0057] Figure 8 An example diagram of the accuracy (left) and error (right) of a CNN model provided by an embodiment of the present invention is shown;

[0058] Fig. 9 An example diagram of the accuracy (left) and error (right) of the target model provided by an embodiment of the present invention is shown;

[0059] Fig.10 The model parameter management interface of a granite rock identification system based on multiple features provided by an embodiment of the present invention is shown;

[0060] Fig.11 The present invention shows a training set supplement interface of a granite rock recognition system based on multiple features provided by an embodiment of the present invention;

[0061] Fig.12 The invention shows an identification result management interface of a granite rock identification system based on multiple features provided by an embodiment of the present invention. DETAILED DESCRIPTION

[0062] The present invention will be described in detail below with reference to the accompanying drawings and in combination with embodiments. It should be noted that the embodiments and features in the embodiments of the present invention can be combined with each other without conflict.

[0063] This embodiment provides a granite rock identification method based on multiple features, such as Figure 1 As shown, the method includes:

[0064] 101. Obtain a rock image to be trained, and perform multi-level feature labeling on the rock image to be trained to obtain a labeled rock image.

[0065] It should be noted that there are multiple rock images to be trained, which may include non-rock images. In order to improve the accuracy of automatic recognition, it is necessary to express geological features at multiple levels. Therefore, each rock image to be trained is labeled with multi-level features. For this embodiment, the basic form of labeling can be: ABCD-..., one letter represents one level, and the number of levels can be set according to the actual application scenario, which is not limited here. There is an order between the letters, which is called the feature level order. The features in each level are called each level features, which are represented by 0, 1, 2...

[0066] 102. Construct an initial model corresponding to each level of feature recognition, and use the annotated rock image to train and test the initial model to obtain a target model corresponding to each level of feature recognition.

[0067] For this embodiment, an initial model corresponds to identifying what the features in a level are. Therefore, during training and testing, a corresponding initial model is trained using the features in a level to obtain a corresponding target model.

[0068] 103. Obtain a rock image to be identified, and identify the rock image to be identified in sequence according to the target model corresponding to the feature level sequence to obtain target multi-level features of the rock image to be identified.

[0069] For this embodiment, there are multiple rock images to be identified. Since the feature levels are in order, the rock images to be identified are identified in sequence according to the target model corresponding to the feature level order. After one identification, the first-level feature is obtained. Until all the rock images to be identified are identified, the target multi-level feature of each rock image to be identified is obtained.

[0070] The present invention provides a method, system, device and equipment for identifying granite rocks based on multiple features. First, a rock image to be trained is obtained, and the rock image to be trained is labeled with multi-level features to obtain a labeled rock image; then an initial model corresponding to each level of feature recognition is constructed, and the initial model is trained and tested using the labeled rock image to obtain a target model corresponding to each level of feature recognition; finally, a rock image to be recognized is obtained, and the rock image to be recognized is sequentially recognized according to the target model corresponding to the feature level sequence to obtain the target multi-level features of the rock image to be recognized. Through the technical solution of the present invention, the rock image to be trained is labeled with multi-level features, and the geological characteristics of the rock itself are fully considered. For each level of feature, an initial model is trained to obtain a target model, and the target model is used to specifically recognize the level of features of the rock image to be recognized, thereby improving the recognition accuracy and improving the recognition efficiency compared to manual recognition.

[0071] Furthermore, as a refinement and extension of the specific implementation of the above embodiment, in order to fully illustrate the specific implementation process of this embodiment, another granite rock identification method based on multiple features is provided, such as Figure 2 As shown, the method includes:

[0072] 201. Obtain a rock image to be trained, determine multi-level features, and label the features of the rock image to be trained level by level to obtain a labeled rock image.

[0073] It should be noted that there are multiple rock images to be trained, including all types of rocks, namely the three typical types of rocks: sedimentary rocks, igneous rocks and metamorphic rocks. These images can be collected from the field or captured on the Internet. In order to improve the accuracy of automatic recognition, it is necessary to express geological features at multiple levels. Therefore, each rock image to be trained is labeled with multi-level features. For this embodiment, the basic form of labeling can be: ABCD-..., one letter represents one level, and the number of levels can be set according to the actual application scenario, which is not limited here. There is an order between the letters, which is called the feature level order. The features in each level are called each level features, which are represented by 0, 1, 2...

[0074] Among them, with the ultimate goal of identifying the lithology of granite, four levels can be designed, and there is an order between the levels, that is, the feature level order. Specifically, the multi-level features include:

[0075] First level A: Rock and non-rock are the first level features, where rock in the first level features is represented by 1 and non-rock is represented by 0;

[0076] Level 2 B: igneous rocks and non-igneous rocks are second-level features, where igneous rocks in the second-level features are represented by 1 and non-igneous rocks are represented by 0;

[0077] Level 3 C: Granite and non-granite are the third-level features, where granite is represented by 1 and non-granite is represented by 0;

[0078] Level 4D: The lithology of granite is the fourth level characteristic, where the lithology is specifically represented by 1 for alkali granite, 2 for potassium granite, 3 for monzogranite, 4 for granodiorite, and 5 for plagiogranite.

[0079] The order of feature levels is reflected in that the rocks in the first-level features include igneous rocks and non-igneous rocks, the igneous rocks in the second-level features include granites and non-granites, and the granites in the third-level features include granite lithologies.

[0080] For example, the complete label for granodiorite is 1-1-1-4; the complete label for peridotite is 1-1-0-0; and the complete label for sandstone is 1-0-0-0. If different rocks appear in an image, the four-level label ABCD can be used repeatedly and cyclically to represent different lithologies at different locations in the same image. For example: 1-1-1-4, 1-0-0-0, which means that two kinds of rocks appear in the image at the same time, the first one is granodiorite (1-1-1-4), and the second one is sandstone (1-0-0-0).

[0081] Preferably, after the rock image to be trained is labeled with features step by step, data enhancement operations may be performed on it by means of translation, scaling, rotation, etc. to obtain a labeled rock image.

[0082] 202. Construct an initial model corresponding to each level of recognition features.

[0083] For this embodiment, constructing an initial model corresponding to each level of feature identification includes: constructing a first initial model for identifying rocks and non-rocks; constructing a second initial model for identifying igneous rocks and non-igneous rocks; constructing a third initial model for identifying granites and non-granites; constructing a fourth initial model for identifying the lithology of granites; using the first initial model, the second initial model, the third initial model and the fourth initial model as the initial models corresponding to each level of feature identification.

[0084] As an implementation method, the first initial model can be a YoloV5 model with a spatial attention mechanism. As an implementation method, the second initial model can be built based on transfer learning and ResNet18. Specifically, the trained ResNet18 model parameters are loaded, the convolution layer and pooling layer of the ResNet18 model are retained, and the number of output nodes of the fully connected layer is modified to 2. As an implementation method, the third initial model can be built based on transfer learning and ResNet18. Specifically, the trained ResNet18 model parameters are loaded, the convolution layer and pooling layer of the ResNet18 model are retained, and the number of output nodes of the fully connected layer is modified to 2. As an implementation method, the fourth initial model can be built based on transfer learning and ResNet101. Specifically, the trained ResNet101 model parameters are loaded, the convolution layer and pooling layer of the ResNet101 model are retained, and the initial output node number of the fully connected layer (+softmax) is modified to 6 (corresponding to five specific granite lithologies (i.e., alkali feldspar granite, potassium feldspar granite, diorite, granodiorite, plagioclase granite) and other granite rocks). If the number of lithologies changes, it is adjusted accordingly.

[0085] 203. The initial model is trained and tested using the annotated rock image to obtain a target model corresponding to each level of feature recognition.

[0086] For this embodiment, the initial model is trained and tested using the annotated rock image to obtain a target model corresponding to each level of feature recognition, including: the first initial model is trained and tested using the annotated rock image to obtain a first target model corresponding to the first level of feature recognition;

[0087] The annotated rock images are named Dataset-1, and Dataset-1 is randomly divided into a training set and a test set according to a preset ratio (e.g., 8:2), which are used for training and testing the first initial model, respectively, and the trained first target model is saved in .pt format.

[0088] Deleting the image whose first-level feature is non-rock in the labeled rock image to obtain a first deleted image; training and testing the second initial model using the first deleted image to obtain a second target model corresponding to identifying the second-level feature;

[0089] Among them, Dataset 1 includes images of rocks and images of non-rocks. The non-rock images are deleted, and the remaining images are all rock images, which are named Dataset 2 (Dataset-2). Dataset 2 is randomly divided into a training set and a test set according to a preset ratio (for example, 8:2), which are used for training and testing the second initial model respectively. During the training process, the PSO algorithm is used to optimize the hyperparameters of the model, and the trained second target model is saved in .pt format.

[0090] Deleting the image whose second-level feature is non-magmatic rock in the first deleted image to obtain a second deleted image; training and testing the third initial model using the second deleted image to obtain a third target model corresponding to the recognition of the third-level feature;

[0091] Among them, Dataset 2 includes images of igneous rocks and images of non-igneous rocks. The images of non-igneous rocks are deleted, and the remaining images are all igneous rocks. It is named Dataset 3 (Dataset-3). Dataset 3 is randomly divided into a training set and a test set according to a preset ratio (for example, 8:2), which are used for training and testing the third initial model respectively. During the training process, the PSO algorithm is used to optimize the hyperparameters of the model, and the trained third target model is saved in .pt format.

[0092] Deleting the image whose third-level feature is not granite in the second deleted image to obtain a third deleted image; training and testing the fourth initial model using the third deleted image to obtain a fourth target model corresponding to the fourth-level feature;

[0093] Among them, Dataset 3 includes granite images and non-granite images. The non-granite images are deleted, and the remaining images are all granite images, which are named Dataset 4 (Dataset-3). Dataset 4 is randomly divided into a training set and a test set according to a preset ratio (for example, 8:2), which are used for training and testing the fourth initial model respectively. During the training process, the PSO algorithm is used to optimize the hyperparameters of the model, and the trained fourth target model is saved in .pt format.

[0094] The first target model, the second target model, the third target model and the fourth target model are used as target models corresponding to each level of features to be identified.

[0095] 204. Obtain a rock image to be identified, and identify the rock image to be identified in sequence according to the target model corresponding to the feature level sequence to obtain target multi-level features of the rock image to be identified.

[0096] For this embodiment, the target model corresponding to the feature level sequence sequentially identifies the rock image to be identified to obtain the target multi-level features of the rock image to be identified, including: identifying the rock image to be identified according to the first target model to obtain the image of the target rock and the image of the target non-rock in the rock image to be identified;

[0097] Among them, the first target model is used to detect the image of the target rock and the image of the target non-rock in the rock image to be identified. If the image of the target rock does not exist, 0-0-0-0 is directly output. If the image of the target rock exists, its position is detected and each target rock image is automatically cropped out.

[0098] Recognize the image of the target rock according to the second target model to obtain an image of a target igneous rock and an image of a target non-igneous rock in the image of the target rock;

[0099] Among them, the second target model is used to detect the image of the target igneous rock and the image of the target non-igneous rock in the image of the target rock. If the image of the target igneous rock does not exist, 1-0-0-0 is directly output. If the image of the target igneous rock exists, it is screened out.

[0100] Recognize the image of the target igneous rock according to the third target model to obtain a target granite image and a target non-granite image in the image of the target igneous rock;

[0101] The third target model is used to detect the target granite image and the target non-granite image in the target igneous rock image. If the target granite image does not exist, 1-1-0-0 is directly output. If the target granite image exists, it is screened out.

[0102] Recognize the target granite image according to the fourth target model to obtain the target lithology of the target granite image;

[0103] The fourth target model is used to detect the target lithology of the target granite image, and the target lithology is labeled according to the preset label number corresponding to the target lithology.

[0104] The target multi-level features of the rock image to be identified are obtained according to the target rock image, the target non-rock image, the target igneous rock image, the target non-igneous rock image, the target granite image, the target non-granite image and the target lithology.

[0105] The application process of step 204 of the embodiment is as follows: Figure 5 shown.

[0106] In summary, the cyclic multi-level label encoding scheme can be used to accurately express complex rock images in a hierarchical manner. The problem of granite lithology identification in complex environments is decomposed step by step based on a multi-level and multi-layer intelligent identification model (i.e., the first target model, the second target model, the third target model, and the fourth target model), which improves the recognition accuracy.

[0107] After all the rock images to be identified are processed, the target multi-level features of the rock images to be identified are stored. Preferably, the rock images to be identified can be stored in the labeled rock images in step 201 of the embodiment to expand the data set used for training and testing, and iteratively update the first target model, the second target model, the third target model and the fourth target model.

[0108] At present, the general CNN model lacks the ability of adaptive model adjustment and parameter update, and is difficult to cope with the dynamically changing data environment. It lacks consideration of the geological characteristics of the rock itself, and focuses more on general recognition technology. It lacks the intelligent lithology recognition method for granite (and its geological characteristics). Therefore, the recognition accuracy is not high. Specifically, Figure 8 , Fig. 9 The training and testing results of the general CNN model and the method of this embodiment are shown respectively. Even when there are multiple categories and multiple features in the training data, the recognition accuracy of this method for granite can reach 92% during the test, while the recognition accuracy of the general CNN model can only reach 83%.

[0109] The present invention provides a method, system, device and equipment for identifying granite rocks based on multiple features. First, a rock image to be trained is obtained, and the rock image to be trained is labeled with multi-level features to obtain a labeled rock image; then an initial model corresponding to each level of feature recognition is constructed, and the initial model is trained and tested using the labeled rock image to obtain a target model corresponding to each level of feature recognition; finally, a rock image to be recognized is obtained, and the rock image to be recognized is sequentially recognized according to the target model corresponding to the feature level sequence to obtain the target multi-level features of the rock image to be recognized. Through the technical solution of the present invention, the rock image to be trained is labeled with multi-level features, and the geological characteristics of the rock itself are fully considered. For each level of feature, an initial model is trained to obtain a target model, and the target model is used to specifically recognize the level of features of the rock image to be recognized, thereby improving the recognition accuracy and improving the recognition efficiency compared to manual recognition.

[0110] The embodiment of the present invention provides a granite rock recognition system based on multiple features, the system comprising: a rock image management module to be trained, a recognition module, and a model management module; the rock image management module to be trained is used to respond to a management operation request of a user for the rock image to be trained, and perform corresponding management operations on the rock image to be trained, wherein the management operation request includes setting the number and type of the rock images to be trained, and adding, deleting, modifying, and querying the rock images to be trained; the recognition module is used to receive the rock images to be identified, and sequentially identify the rock images to be identified according to the target models corresponding to the feature level sequence, and obtain the target multi-level features of the rock images to be identified; the model management module is used to re-train the target model when the number of the rock images to be trained that are newly added or modified by the user is greater than a preset threshold, and during the training, receive hyperparameters or automatically search for the hyperparameters to obtain an updated target model, so that the recognition model receives the rock images to be identified, and sequentially identifies the rock images to be identified according to the updated target models corresponding to the feature level sequence, and obtains the updated target multi-level features of the rock images to be identified.

[0111] It should be noted that Figure 6 This is a system architecture diagram, which mainly includes: Android-based mobile terminal (front end), cloud server end, system backend and database. Among them, the front end is responsible for the functional interface and user interaction, the back end is responsible for front end requests, manages related data, models and other resources, accesses the database in an orderly manner, and sends the corresponding results, data, resources, etc. to the front end. The database is responsible for storing information, including user information, model information, etc. The cloud server is responsible for providing public network IP, realizing request forwarding, and facilitating request sending.

[0112] It should be noted that Figure 7 This is a system function diagram. From the perspective of logical structure, the system can be divided into six layers: App display layer, App business logic layer, cloud server layer, backend server layer and resource support layer. Figure 10-12 Shows the typical functional interface of this system.

[0113] The core functional modules of the system include: training rock image management module, recognition module, and model management module. Specifically, the training rock image management module provides users with an interface for adding, deleting, modifying, and checking training data in the system database, allowing users to set the number and type of training data. The recognition module uses the Android system's photo album and camera to obtain rock images, uses http request encapsulation technology to send requests, and the backend calls the first target model, the second target model, the third target model, and the fourth target model for lithology identification, and returns the identification results. When the number of images added or adjusted by the user reaches a certain threshold, the model management module triggers re-model training, and supports manual adjustment or PSO automatic search of model hyperparameters. The model structure and parameters can be dynamically updated as the application scenario changes.

[0114] Further, as Figure 1 and Figure 2 The specific implementation of the method shown in the figure, the embodiment of the present invention provides a granite rock identification device based on multiple features, such as Figure 3 As shown, the device includes: an acquisition module 31, a training module 32, and an application module 33;

[0115] An acquisition module 31 is used to acquire a rock image, and label the rock image with multi-level features to obtain a labeled rock image;

[0116] The training module 32 is used to construct an initial model corresponding to each level of feature recognition, and to train and test the initial model using the annotated rock image to obtain a target model corresponding to each level of feature recognition;

[0117] The application module 33 is used to obtain the rock image to be identified, identify the rock image to be identified in sequence according to the target model corresponding to the feature level sequence, and obtain the target multi-level features of the rock image to be identified.

[0118] Accordingly, in order to label the rock image to be trained with multi-level features and obtain the labeled rock image, the acquisition module 31 includes a determination unit 311 and a labeling unit 312;

[0119] The determining unit 311 is specifically used to determine a multi-level feature, wherein the multi-level feature includes: rock and non-rock as a first-level feature, igneous rock and non-igneous rock as a second-level feature, granite and non-granite as a third-level feature, and the lithology of granite as a fourth-level feature;

[0120] The labeling unit 312 is specifically used to label the features of the rock image to be trained step by step to obtain a labeled rock image.

[0121] Accordingly, in order to construct an initial model corresponding to each level of feature identification, the training module 32 is specifically used to construct a first initial model for identifying rocks and non-rocks; construct a second initial model for identifying igneous rocks and non-igneous rocks; construct a third initial model for identifying granites and non-granites; construct a fourth initial model for identifying the lithology of granites; and use the first initial model, the second initial model, the third initial model and the fourth initial model as the initial models corresponding to each level of feature identification.

[0122] Accordingly, in order to use the annotated rock image to train and test the initial model respectively and obtain the target model corresponding to each level of feature recognition, the training module 32 is specifically used to use the annotated rock image to train and test the first initial model and obtain the first target model corresponding to the first level of feature recognition; delete the image of the first level feature of the annotated rock image as non-rock to obtain the first deleted image; use the first deleted image to train and test the second initial model and obtain the second target model corresponding to the second level of feature recognition; delete the image of the second level feature of the first deleted image as non-magmatic rock to obtain the second deleted image; use the second deleted image to train and test the third initial model and obtain the third target model corresponding to the third level of feature recognition; delete the image of the third level feature of the second deleted image as non-granite to obtain the third deleted image; use the third deleted image to train and test the fourth initial model and obtain the fourth target model corresponding to the fourth level of feature recognition; use the first target model, the second target model, the third target model and the fourth target model as the target models corresponding to each level of feature recognition.

[0123] Accordingly, in order to identify the rock image to be identified in sequence according to the target models corresponding to the feature level order and obtain the target multi-level features of the rock image to be identified, the application module 33 is specifically used to identify the rock image to be identified according to the first target model to obtain the image of the target rock and the image of the target non-rock in the rock image to be identified; identify the image of the target rock according to the second target model to obtain the image of the target igneous rock and the image of the target non-igneous rock in the image of the target rock; identify the image of the target igneous rock according to the third target model to obtain the image of the target granite and the image of the target non-granite in the image of the target igneous rock; identify the image of the target granite according to the fourth target model to obtain the target lithology of the target granite image; obtain the target multi-level features of the rock image to be identified according to the image of the target rock, the image of the target non-rock, the image of the target igneous rock, the image of the target non-igneous rock, the image of the target granite, the image of the target non-granite and the target lithology.

[0124] It should be noted that for other corresponding descriptions of the functional units involved in the granite rock identification device based on multiple features provided in this embodiment, reference can be made to Figure 1 to Figure 2 The corresponding description will not be repeated here.

[0125] Based on the above Figure 1 to Figure 2 The method shown in the embodiment, accordingly, also provides a storage medium, which can be volatile or non-volatile, and stores a computer program, which is executed by a processor to implement the above-mentioned Figure 1 to Figure 2 The multi-feature based granite rock identification method shown.

[0126] Based on this understanding, the technical solution of the present invention can be embodied in the form of a software product, which can be stored in a storage medium (which can be a CD-ROM, a USB flash drive, a mobile hard disk, etc.), and includes a number of instructions for enabling a computer device (which can be a personal computer, a server, or a network device, etc.) to execute the methods of various implementation scenarios of the present invention.

[0127] Based on the above Figure 1 to Figure 2 The method shown and Figure 3 , Figure 4 In order to achieve the above-mentioned purpose, the present embodiment further provides a computer device, which includes a storage medium and a processor; the storage medium is used to store a computer program; the processor is used to execute the computer program to achieve the above-mentioned Figure 1 to Figure 2 The multi-feature based granite rock identification method shown.

[0128] Optionally, the computer device may further include a user interface, a network interface, a camera, a radio frequency (RF) circuit, a sensor, an audio circuit, a WI-FI module, etc. The user interface may include a display, an input unit such as a keyboard, etc., and the optional user interface may also include a USB interface, a card reader interface, etc. The network interface may optionally include a standard wired interface, a wireless interface (such as a WI-FI interface), etc.

[0129] Those skilled in the art will appreciate that the computer device structure provided in this embodiment does not constitute a limitation on the physical device, and may include more or fewer components, or a combination of certain components, or different component arrangements.

[0130] The storage medium may also include an operating network communication module. The operating is a program that manages the hardware and software resources of the above-mentioned computer device, supporting the operation of the information processing program and other software and / or programs. The network communication module is used to communicate between the components inside the storage medium, and to communicate with other hardware and software in the information processing entity device.

[0131] Through the description of the above implementation modes, those skilled in the art can clearly understand that the present invention can be implemented by means of software plus a necessary general hardware platform, or by hardware.

[0132] The present invention provides a method, system, device and equipment for identifying granite rocks based on multiple features. First, a rock image to be trained is obtained, and the rock image to be trained is labeled with multi-level features to obtain a labeled rock image; then an initial model corresponding to each level of feature recognition is constructed, and the initial model is trained and tested using the labeled rock image to obtain a target model corresponding to each level of feature recognition; finally, a rock image to be recognized is obtained, and the rock image to be recognized is sequentially recognized according to the target model corresponding to the feature level sequence to obtain the target multi-level features of the rock image to be recognized. Through the technical solution of the present invention, the rock image to be trained is labeled with multi-level features, and the geological characteristics of the rock itself are fully considered. For each level of feature, an initial model is trained to obtain a target model, and the target model is used to specifically recognize the level of features of the rock image to be recognized, thereby improving the recognition accuracy and improving the recognition efficiency compared to manual recognition.

[0133] Those skilled in the art will appreciate that the accompanying drawings are only schematic diagrams of a preferred implementation scenario, and the modules or processes in the accompanying drawings are not necessarily necessary for implementing the present invention. Those skilled in the art will appreciate that the modules in the devices in the implementation scenario can be distributed in the devices of the implementation scenario according to the description of the implementation scenario, or can be changed accordingly and located in one or more devices different from the present implementation scenario. The modules of the above-mentioned implementation scenario can be combined into one module, or can be further split into multiple submodules.

[0134] The above serial numbers of the present invention are only for description and do not represent the advantages and disadvantages of the implementation scenarios. The above disclosure is only a few specific implementation scenarios of the present invention, but the present invention is not limited thereto, and any changes that can be thought of by those skilled in the art should fall within the protection scope of the present invention.

Claims

1. A multi-feature based granite rock identification method, characterized in that: The method comprises: Acquire a rock image to be trained, and perform multi-level feature labeling on the rock image to be trained to obtain a labeled rock image; Constructing an initial model corresponding to each level of feature recognition, and using the annotated rock images to train and test the initial model respectively, to obtain a target model corresponding to each level of feature recognition; The rock image to be identified is acquired, and the rock image to be identified is identified in sequence according to the target model corresponding to the feature level sequence to obtain the target multi-level features of the rock image to be identified.

2. The method according to claim 1, characterized in that The step of labeling the to-be-trained rock image with multi-level features to obtain a labeled rock image includes: Determine a multi-level feature, wherein the multi-level feature includes: rock and non-rock as a first-level feature, igneous rock and non-igneous rock as a second-level feature, granite and non-granite as a third-level feature, and lithology of granite as a fourth-level feature; The rock image to be trained is labeled with features step by step to obtain a labeled rock image.

3. The method according to claim 2, characterized in that The construction of the initial model corresponding to each level of feature recognition includes: constructing a first initial model for identifying rocks and non-rocks; constructing a second initial model for identifying igneous rocks and non-igneous rocks; Constructing a third initial model for identifying granitoids and non-granitoids; constructing a fourth initial model for identifying lithology of granitoids; The first initial model, the second initial model, the third initial model and the fourth initial model are used as initial models corresponding to each level of feature identification.

4. The method according to claim 3, characterized in that: The initial model is trained and tested using the annotated rock image to obtain a target model corresponding to each level of feature recognition, including: Using the annotated rock image to train and test the first initial model, to obtain a first target model corresponding to identifying the first level features; Deleting the image whose first-level feature is non-rock in the labeled rock image to obtain a first deleted image; Using the first deleted image to train and test the second initial model, to obtain a second target model corresponding to identifying the second level characteristics; Deleting the image whose second-level feature is non-magmatic rock in the first deleted image to obtain a second deleted image; Using the second deleted image to train and test the third initial model, to obtain a third target model corresponding to the recognition of the third level characteristics; Deleting the image whose third-level feature is not granite in the second deleted image to obtain a third deleted image; Using the third deleted image to train and test the fourth initial model, to obtain a fourth target model corresponding to the fourth level feature recognition; The first target model, the second target model, the third target model and the fourth target model are used as target models corresponding to each level of features to be identified.

5. The method according to any one of claims 2 to 4, characterized in that: The target model corresponding to the feature level sequence sequentially identifies the rock image to be identified to obtain the target multi-level features of the rock image to be identified, including: Identify the rock image to be identified according to the first target model to obtain an image of a target rock and an image of a target non-rock in the rock image to be identified; Recognize the image of the target rock according to the second target model to obtain an image of a target igneous rock and an image of a target non-igneous rock in the image of the target rock; Recognize the image of the target igneous rock according to the third target model to obtain a target granite image and a target non-granite image in the image of the target igneous rock; Recognize the target granite image according to the fourth target model to obtain the target lithology of the target granite image; The target multi-level features of the rock image to be identified are obtained according to the target rock image, the target non-rock image, the target igneous rock image, the target non-igneous rock image, the target granite image, the target non-granite image and the target lithology.

6. A granite rock identification system based on multiple features, characterized in that: The system comprises: a rock image management module to be trained, a recognition module, and a model management module; The to-be-trained rock image management module is used to respond to a user's request for a management operation on the to-be-trained rock image, and perform a corresponding management operation on the to-be-trained rock image, wherein the management operation request includes setting the number and type of the to-be-trained rock images, and adding, deleting, modifying, and querying the to-be-trained rock images; The recognition module is used to receive the rock image to be recognized, and recognize the rock image to be recognized in sequence according to the target model corresponding to the feature level sequence, so as to obtain the target multi-level features of the rock image to be recognized; The model management module is used to re-train the target model when the number of rock images to be trained newly added or modified by the user is greater than a preset threshold, and during training, receive hyperparameters or automatically search for the hyperparameters to obtain an updated target model, so that the recognition model receives the rock images to be recognized, and recognizes the rock images to be recognized in turn according to the updated target models corresponding to the feature level sequence, to obtain updated target multi-level features of the rock images to be recognized.

7. A granite rock identification device based on multiple features, characterized in that: The device comprises: An acquisition module is used to acquire a rock image, and label the rock image with multi-level features to obtain a labeled rock image; A training module is used to construct an initial model corresponding to each level of feature recognition, and to train and test the initial model using the annotated rock image to obtain a target model corresponding to each level of feature recognition; The application module is used to obtain the rock image to be identified, identify the rock image to be identified in turn according to the target model corresponding to the feature level sequence, and obtain the target multi-level features of the rock image to be identified.

8. The device according to claim 7, characterized in that The acquisition module includes a determination unit and a marking unit; The determination unit is used to determine a multi-level feature, wherein the multi-level feature includes: rock and non-rock as a first-level feature, igneous rock and non-igneous rock as a second-level feature, granite and non-granite as a third-level feature, and the lithology of granite as a fourth-level feature; The labeling unit is used to label the features of the rock image to be trained step by step to obtain a labeled rock image.

9. A storage medium having a computer program stored thereon, characterized in that: When the computer program is executed by a processor, the multi-feature based granite rock identification method described in any one of claims 1 to 5 is implemented.

10. A computer device comprising a storage medium, a processor, and a computer program stored in the storage medium and executable on the processor, characterized in that: When the processor executes the computer program, the granite rock identification method based on multiple features described in any one of claims 1 to 5 is implemented.

Images