Marine litter fine granularity identification method and system based on unmanned aerial vehicle image
By constructing a marine debris identification method that employs incremental learning tasks and adaptively adjusts the model structure, the problems of catastrophic forgetting and insufficient fine-grained discrimination in marine debris identification are solved, achieving efficient and stable identification of marine debris.
Patent Information
- Authority / Receiving Office
- CN · China
- Patent Type
- Applications(China)
- Current Assignee / Owner
- HANGZHOU DIANZI UNIV
- Filing Date
- 2026-03-31
- Publication Date
- 2026-05-01
AI Technical Summary
Existing incremental target detection methods suffer from catastrophic forgetting problems and insufficient fine-grained discrimination capabilities in marine debris identification tasks, making it difficult to cope with complex fine-grained scenarios, and lacking dynamic modeling of the similarity and differentiation difficulty of different categories.
A fine-grained marine debris identification method based on UAV imagery is adopted. By constructing a class-incremental learning task, using a prior analyzer to calculate the task similarity index, adaptively adjusting the model structure, enabling or bypassing RoI fine-grained discrimination modulation, and combining attention mechanism weighted modulation features, the fine-grained discrimination capability is enhanced.
It effectively alleviates the problem of fine-grained knowledge forgetting in the incremental learning process, improves the long-term identification stability and fine-grained discrimination ability of marine debris categories, and enhances identification efficiency and coverage.
Smart Images

Figure CN121963002A_ABST
Abstract
Description
A method and system for fine-grained identification of marine debris based on UAV imagery Technical Field
[0001] This invention belongs to the field of computer vision and deep learning technology, specifically relating to a method and system for fine-grained identification of marine debris based on UAV imagery. Background Technology
[0002] With the rapid development of UAV remote sensing technology, intelligent sensing, and deep learning technology, automatic detection and identification of marine debris based on UAV imagery has become an important technical means in marine environmental monitoring, pollution control, and emergency response. Compared with manual inspection methods, deep learning-based target detection models can significantly improve identification efficiency and coverage in large-scale, high-frequency marine monitoring tasks.
[0003] However, in real-world applications, the classification system for marine debris is not static. On the one hand, as monitoring time extends, new types of debris constantly emerge, such as new types of plastic products, composite material debris, and special debris generated by unforeseen events. On the other hand, the distribution of debris categories varies significantly across different regions and sea conditions. This necessitates that marine debris identification systems possess the ability to continuously learn new categories, rather than relying on fixed models trained offline only once.
[0004] To address the aforementioned issues, incremental object detection methods have been proposed. Their core objective is to enable the model to progressively learn new object categories based on existing categories without revisiting all historical training data. However, existing incremental object detection methods still face the following shortcomings in marine debris identification tasks:
[0005] First, in incremental class learning, models commonly suffer from catastrophic forgetting, meaning that the detection performance of already learned categories significantly decreases when learning new categories. Although existing methods mitigate this overall forgetting problem through knowledge distillation, sample replay, or regularization constraints, they primarily focus on maintaining category-level detection performance and are insufficient to adequately address complex, fine-grained scenarios. Second, marine debris identification tasks exhibit significant fine-grained recognition characteristics. Different debris categories are often highly similar in appearance, material texture, and local structure, such as different types of plastic buoys, foam products, or composite material debris. These fine-grained differences are typically distinguished by local texture, edge structure, or subtle morphological variations, placing higher demands on the model's feature representation capabilities. In incremental class learning, when learning new categories, models tend to prioritize adapting to new discrimination boundaries, gradually weakening their focus on fine-grained discriminative features between already learned categories. This leads to implicit forgetting of fine-grained discriminative knowledge between learned categories. This problem differs from traditional category-level forgetting; it often manifests as categories still being detectable, but misclassification between them significantly increases, severely impacting the long-term stability of the system. In addition, most existing methods use fixed thresholds or static strategies to determine whether to introduce additional modules, lacking dynamic modeling of class similarity and discrimination difficulty between different classes of incremental learning tasks. This makes it difficult to enhance fine-grained discrimination capabilities while ensuring model simplicity.
[0006] Therefore, there is an urgent need in this field for a marine debris identification method that can adaptively adjust the model structure within the framework of incremental target detection, combined with inter-task category similarity analysis, and effectively alleviate the problem of knowledge forgetting in fine-grained discrimination. Summary of the Invention
[0007] Based on the aforementioned shortcomings and deficiencies in the prior art, one of the objectives of this invention is to at least solve one or more of the aforementioned problems in the prior art. In other words, one of the objectives of this invention is to provide a method and system for fine-grained identification of marine debris based on UAV imagery that meets one or more of the aforementioned requirements.
[0008] To achieve the above-mentioned objectives, the present invention adopts the following technical solution: a fine-grained identification method for marine debris based on UAV imagery, comprising the following steps: S1, constructing a target detection dataset based on UAV data and dividing it into N incremental learning tasks; wherein, the target detection dataset contains different categories of marine debris, and N is an integer greater than 1; S2, sequentially inputting the incremental learning tasks into the target detection model for training to obtain the trained target detection model; wherein, during the training of the current (t+1)th incremental learning task, samples are extracted from the data corresponding to the historical (t)th incremental learning tasks and the current (t+1)th incremental learning task to form prior analysis. The dataset and prior analysis dataset are input into the prior analyzer for similarity analysis to obtain a task similarity index that characterizes the difficulty of distinguishing marine debris categories; t∈[1,N-1]; based on the task similarity index, it is determined whether to enable RoI fine-grained discrimination modulation during the training of the current (t+1)th class incremental learning task; if yes, the RoI features in the target detection model training process are weighted and modulated based on the attention mechanism and then the original network structure is continued; if no, the original network structure of the target detection model is maintained for training; S3, collect the target detection data to be identified through UAV, input it into the trained target detection model, and output the target detection results.
[0009] As a preferred embodiment, step S1 specifically includes the following steps: S11, collecting a target detection dataset containing multiple marine debris categories based on UAVs; wherein each data sample includes at least an input image and the corresponding target bounding box and category label; S12, dividing the target detection dataset into multiple non-overlapping category subsets according to all marine debris categories; S13, each category subset corresponds to a class incremental learning task, and is sequentially input into the target detection model for training in a predetermined order.
[0010] As a preferred embodiment, step S2, the process of inputting the prior analysis dataset into the prior analyzer for similarity analysis includes the following steps: S21, inputting the prior analysis dataset into the target detection model that has completed the training of the t-th class incremental learning task, and extracting the feature vector corresponding to each target instance from the regional interest feature layer of the model; S22, aggregating the feature vectors of target instances belonging to the same marine debris category to obtain category feature prototypes; S23, calculating the intra-task similarity between each category feature prototype in the current (t+1)-th class incremental learning task; calculating the cross-task similarity between each category feature prototype in the current (t+1)-th class incremental learning task and each category feature prototype in the historical t-th class incremental learning tasks; S24, accumulating the intra-task similarity and cross-task similarity and normalizing them to obtain the task similarity index.
[0011] As a preferred option, the task similarity index of the (t+1)th class incremental learning task for: ;in, This represents the sum of similarities within each task. It is the sum of cross-task similarity. This represents the total number of category feature prototype pairs involved in the similarity calculation.
[0012] As a preferred embodiment, step S2, determining whether RoI fine-grained discriminative modulation needs to be enabled during the training of the current (t+1)th class incremental learning task based on the task similarity index, includes the following steps: S25, determining whether t is 1; if yes, the current class incremental learning task is the second class incremental learning task, proceed to step S26; if no, proceed to step S27; S26, determining whether the task similarity index of the second class incremental learning task is greater than the task similarity index of the first class incremental learning task; if yes, RoI fine-grained discriminative modulation is enabled; if no, RoI fine-grained discriminative modulation is not enabled; S27, calculating the distance between the task similarity index of the current class incremental learning task and the task similarity indices of the historical t class incremental learning tasks, determining the target historical class incremental learning task that is closest to the current class incremental learning task; and determining whether RoI fine-grained discriminative modulation is enabled during the training of the current class incremental learning task based on the enabled status of RoI fine-grained discriminative modulation of the target historical class incremental learning task during the training process.
[0013] As a preferred embodiment, the task similarity index of the current incremental learning task is compared with that of the historical task. The distance between the task similarity metrics of each class of incremental learning tasks is: ;in, .
[0014] As a preferred embodiment, the target historical incremental learning task is the first... Incremental learning task for each class: .
[0015] As a preferred embodiment, the weighted modulation process based on the attention mechanism includes: first, global pooling of the RoI features to obtain channel statistical vectors that reflect the strength of each channel response; then, setting attention branches for different types of incremental learning tasks during the activation phase; generating channel weight coefficients based on the channel statistical vectors by the attention branches of the corresponding tasks; and performing weighted modulation of the RoI features in a residual manner.
[0016] As a preferred option, the object detection model is the object detection framework Faster R-CNN.
[0017] This invention also provides a fine-grained marine debris identification system based on UAV imagery, applying the fine-grained marine debris identification method described in any of the preceding solutions. The fine-grained marine debris identification system includes: a construction module for constructing a target detection dataset based on UAV data and dividing it into N incremental learning tasks; wherein the target detection dataset contains different marine debris categories, and N is an integer greater than 1; and a training module for sequentially inputting the incremental learning tasks into the target detection model for training, obtaining the trained target detection model; wherein, during the training of the current (t+1)th incremental learning task, data is collected based on the data corresponding to the historical (t)th incremental learning tasks and the current (t+1)th incremental learning task. Samples are extracted to form a prior analysis dataset, which is then input into a prior analyzer for similarity analysis to obtain a task similarity index that characterizes the difficulty of distinguishing marine debris categories; t∈[1,N-1]; based on the task similarity index, it is determined whether to enable RoI fine-grained discrimination modulation during the training of the current (t+1)th class incremental learning task; if yes, the RoI features in the target detection model training process are weighted and modulated based on an attention mechanism before continuing the processing of the remaining network structures; if no, the original network structure of the target detection model is maintained for training; the detection module is used to collect target detection data to be identified through UAVs, input it into the trained target detection model, and output the target detection results.
[0018] Compared with the prior art, the beneficial effects of this invention are: this invention can enhance the fine-grained discrimination ability in the incremental stage where the distinction between marine debris categories is more difficult, and avoid unnecessary structural disturbances in the stage where the distinction between marine debris categories is less difficult, thereby effectively alleviating the problem of fine-grained knowledge forgetting in the class incremental learning process and improving the long-term recognition stability of similar marine debris categories. Attached Figure Description
[0019] Figure 1 is a schematic diagram of the construction of the prior analysis dataset in Embodiment 1 of the present invention; Figure 2 is a schematic diagram of the processing flow of the prior analyzer in Embodiment 1 of the present invention; Figure 3 is a schematic diagram of the decision mechanism without fixed threshold in Embodiment 1 of the present invention; Figure 4 is a flowchart of the detection process of fine-grained discrimination modulation enable / bypass of RoI guided by prior decision in Embodiment 1 of the present invention. Detailed Implementation
[0020] To more clearly illustrate the embodiments of the present invention, specific implementation methods will be described below with reference to the accompanying drawings. Obviously, the drawings described below are merely some embodiments of the present invention. For those skilled in the art, other drawings and other implementation methods can be obtained based on these drawings without any creative effort.
[0021] This invention is based on the object detection framework Faster R-CNN and includes a class incremental training data construction module, a prior similarity analysis module, a decision module without a fixed threshold, a RoI fine-grained discriminative modulation module, and an object detection output module. The prior similarity analysis module is used to characterize the difficulty of class discrimination between different class incremental learning tasks. The decision module without a fixed threshold is used to adaptively determine whether to enable or bypass the RoI fine-grained discriminative modulation module based on the difficulty, thereby achieving dynamic structure adjustment without introducing a fixed threshold.
[0022] First, marine debris images collected by UAVs are input into the target detection model, and multi-level feature representations are extracted through a backbone feature extraction network. Then, according to a predetermined category introduction order, marine debris categories are divided into multiple consecutive incremental learning tasks, and the model is trained stage by stage according to the task order. After the training of each incremental learning task is completed and before the training of the next incremental learning task, a subset of samples is extracted from the data corresponding to the previously learned task and the data corresponding to the current task. Based on the Region Interest (RoI) features of the target instance, feature prototypes of each category are constructed, and the similarity relationship between category prototypes is calculated to obtain a task similarity index used to characterize the overall category discrimination difficulty of the current task. The task similarity index comprehensively considers the internal category similarity of the current task and the cross-task similarity between the current task and historical tasks, and can reflect the potential fine-grained confusion degree in subsequent incremental training. Based on this, the present invention employs a relative comparison decision mechanism without a fixed threshold, adaptively determining whether to enable the RoI fine-grained discriminative modulation module according to the task similarity index: when the decision result indicates that it should be enabled, the RoI fine-grained discriminative modulation module is enabled in the RoI feature processing stage of the target detection model, and the RoI features are weighted and modulated based on an attention mechanism to enhance the fine-grained discriminative feature response; when the decision result indicates that it should not be enabled, the RoI fine-grained discriminative modulation module is bypassed, keeping the rest of the target detection model unchanged to avoid unnecessary increases in model complexity and training perturbations. In addition, a task-level feature modulation module (i.e., weighted modulation based on an attention mechanism) can be set at the high-level feature output position of the backbone feature extraction network and kept enabled during training to enhance the stability of feature representation. Finally, after enabling or bypassing the RoI fine-grained discriminative modulation module, the current class incremental learning task continues to be trained, and the detection results containing target category and location information are output, thereby achieving continuous learning and stable identification of marine debris targets.
[0023] Example 1: The fine-grained marine debris identification method based on UAV imagery in this example includes the following steps: (1) Constructing a UAV marine debris target detection training dataset and dividing the training data into N consecutive class incremental learning tasks according to a predetermined category introduction order, providing a task sequence and data foundation for subsequent stage-by-stage continuous learning and training; where N is an integer greater than 1; using the publicly available marine floating debris target detection dataset Sea Trash, Sea The Trash training dataset consists of 6194 images, with 783 for the test set and 771 for the validation set. Each dataset is 1200×800 pixels in size and covers various types of marine debris, including plastic buoys, foam buoys, plastic bottles, foam fragments, and metal debris. First, a target detection dataset containing multiple target categories is acquired using a drone. Each sample includes at least the input image, the corresponding target bounding box, and the category label. Second, the dataset is partitioned, dividing all target categories into multiple non-overlapping subsets. Then, multiple consecutive class incremental learning tasks are constructed based on these subsets. Each class incremental learning task includes only the target categories newly added in the current stage, excluding target categories from subsequent stages. Let the set of categories included in the t-th class incremental learning task be: ;in, Let be the total number of categories included in the incremental learning task for the t-th class. For any ,satisfy: Finally, the dataset corresponding to each class incremental learning task is treated as an independent training stage and input into the object detection model in a predetermined order for training.
[0024] (2) Input the incremental learning tasks of each class into the target detection model for training in sequence; before completing the training of the t-th incremental learning task and entering the training of the t+1-th incremental learning task, extract a subset of samples from the historical task and current task data and perform prior similarity analysis to obtain prior analysis results that characterize the difficulty of class distinction in the current task. The specific process is as follows: As shown in Figure 1, the historical task dataset that has been learned includes historical task data 1, historical task data 2, ..., historical task data t. The dataset for the current t+1-th incremental learning task is the current task data t+1. The future task data that has not yet been learned does not participate in the current stage analysis. Based on the historical task dataset and the current task data t+1, a subset of samples is extracted. The samples constitute the prior analysis dataset. Sampling methods can be random sampling or stratified sampling by category. The sampling ratio is determined by preset parameters, preferably a fixed proportion of the total number of samples in each task dataset. This prior analysis dataset is only used for similarity analysis and does not participate in the model parameter update process during the training phase of the (t+1)th class incremental learning task. Future task data also does not participate in the sampling and analysis in this stage. The prior analysis dataset is input into the prior analyzer for similarity analysis, which calls the already trained target detection model for forward inference. As shown in Figure 2, feature vectors corresponding to each target instance are extracted through RoI features to form a set of target instance feature vectors. Let the feature vector of the nth target instance in class n be denoted as: The feature vectors of target instances belonging to the same category are aggregated to obtain the feature prototype of that category. In this embodiment, mean aggregation is used to obtain the feature prototype of the Nth category as follows: ;in, This represents the number of target instances of class 𝑐.
[0025] Then, calculate the internal similarity between feature prototypes of each category in the current task, and the cross-task similarity between feature prototypes of each category in the current task and feature prototypes of each category in historical tasks.
[0026] Specifically, cosine similarity is used to measure the similarity between any two class prototypes. The specific formula for calculating cosine similarity can be found in existing technologies and will not be elaborated here. Finally, the intra-task similarity and cross-task similarity are summed, and the results are normalized according to the classes involved in the similarity calculation to obtain the task similarity index for the current task. ;in, This represents the sum of similarities within each task. It is the sum of cross-task similarity. This represents the total number of category feature prototype pairs participating in the similarity calculation; the task similarity index in this embodiment is used to characterize the overall category discrimination difficulty of the current class incremental learning task.
[0027] (3) Based on the prior analysis results, a relative comparison decision mechanism without a fixed threshold is adopted to determine whether RoI fine-grained discriminative modulation needs to be enabled before entering the training of the (t+1)th class incremental learning task; after obtaining the current task similarity and the historical task similarity, as shown in Figure 3, the task similarity index is used to determine whether RoI fine-grained discriminative modulation needs to be enabled during the training of the current (t+1)th class incremental learning task, including the following steps: (a) Determine whether t is 1; if yes, the current class incremental learning task is the second class incremental learning task, and proceed to step (b); if no, proceed to step (c); (b) Determine the task similarity of the second class incremental learning task. Is the task similarity greater than that of the incremental learning task of the first class? If yes, then enable RoI fine-grained discriminative modulation; if no, then disable RoI fine-grained discriminative modulation, i.e., bypass; since there are no historical tasks during the training of the first class incremental learning task, the task similarity of the first class incremental learning task only considers the internal similarity between the feature prototypes of each class in the first class incremental learning task, that is, the internal similarity of the task is accumulated and normalized to obtain the task similarity of the first class incremental learning task; (c) calculate the distance between the task similarity index of the current class incremental learning task and the task similarity index of the historical t class incremental learning tasks, and determine the target historical class incremental learning task (i.e., the closest historical task) that is closest to the current class incremental learning task; and determine whether the current class incremental learning task enables RoI fine-grained discriminative modulation during the training process based on the enabling status of RoI fine-grained discriminative modulation of the target historical class incremental learning task during the training process; specifically, the task similarity of the current class incremental learning task and the historical t class incremental learning task are calculated. The distance between the task similarities of each class of incremental learning tasks is: ;in, The above-mentioned incremental learning task for historical classes is the first... Incremental learning task for each class: ; Subsequent readings directly from the first... The enabled / bypass state of RoI fine-grained discriminative modulation corresponding to each incremental learning task is determined and used as the enabled / bypass state of the current task. The decision output is the switch control signal of RoI fine-grained discriminative modulation. The above process does not rely on a preset fixed threshold, but achieves adaptive determination based on the relative relationship of historical tasks.
[0028] (4) Based on the decision results, enable or bypass the fine-grained discrimination modulation of RoI in the RoI feature processing stage of the target detection model, enhance the fine-grained discrimination capability when needed, and keep the rest of the model structure unchanged when not needed.
[0029] As shown in Figure 4, the dataset corresponding to the (t+1)th class incremental learning task is input into the object detection model Faster R-CNN. First, a multi-level feature representation is extracted through the backbone feature extraction network. Then, candidate region generation and RoI feature extraction are performed. Next, in the RoI feature processing stage of the object detection model, a RoI fine-grained discriminative modulation enable / bypass switch is set. The control signal for the switch comes from the switch control signal output by the above decision. When the decision result indicates "enable," the switch switches to the enabled path. The RoI features are then weighted and modulated based on an attention mechanism and input into the detection head of the original network. Enhance fine-grained discriminative feature responses; when the decision result indicates "bypass," the switch is switched to the bypass path, allowing RoI features to be directly input into the detection head, thereby keeping the rest of the model structure unchanged and avoiding unnecessary complexity increases; the weighted modulation process based on the attention mechanism in this embodiment includes: first, global pooling of RoI features to obtain channel statistical vectors reflecting the strength of each channel response; then, setting attention branches for different classes of incremental learning tasks during the activation phase; the attention branches of the corresponding tasks generate channel weight coefficients based on the channel statistical vectors, and weighted modulation of RoI features in a residual manner. Unlike conventional attention mechanisms (SE), this embodiment sets the modulation position in the RoI feature processing stage and performs specific task branch processing. Combined with the enable / bypass control of class incremental tasks, instance features are only discriminated and enhanced in stages with high fine-grained confusion risk to reduce unnecessary perturbations and alleviate forgetting; after completing the above enable / bypass settings, training continues for the (t+1)th class incremental learning task, and the detection results are output. The detection results include at least the target category and target localization information, such as bounding boxes and category labels. The above steps are repeated in subsequent incremental learning tasks to achieve continuous learning and stable identification of marine debris targets, resulting in a trained target detection model, which can then be used for fine-grained identification of marine debris.
[0030] (5) Collect target detection data through UAV, input it into the trained target detection model, and output the target detection results.
[0031] The following section verifies the effectiveness of the fine-grained marine debris identification method in this embodiment: Under the incremental target detection task setting of this embodiment, using the same task division, training protocol, and evaluation criteria, the detection performance of the baseline scheme and the identification method of this embodiment are compared and evaluated. The evaluation metric is the average category accuracy of 11 marine debris categories (the average category accuracy obtained by averaging the detection accuracy of the 11 target categories); the baseline scheme uses existing experience playback training.
[0032] Table 1 Comparison of Average Category Precision Under the same incremental target detection task division, training process and evaluation criteria, the detection performance of the baseline method using only the experience replay strategy and the method of the present invention were compared. As shown in Table 1, the average class accuracy of the experience replay method for 11 classes was 24.57%, while the average class accuracy of the method in Example 1 was 51.3%. This shows that the present invention can significantly improve the overall detection and recognition effect of marine debris targets in the incremental learning scenario, especially in the ability to distinguish fine-grained easily confused categories.
[0033] Based on the aforementioned method for fine-grained marine debris identification using UAV imagery, the marine debris fine-grained identification system of this embodiment includes the following functional modules: a construction module, a training module, and a detection module. The construction module constructs a target detection dataset based on data collected by UAVs and divides it into N incremental learning tasks. The training module sequentially inputs the incremental learning tasks into the target detection model for training, resulting in a trained target detection model. Specifically, during the training of the current (t+1)th incremental learning task, samples are extracted from the data corresponding to the historical (t)th incremental learning tasks and the current (t+1)th incremental learning task to form a prior analysis dataset. This prior analysis dataset is then input into a prior analyzer for similarity analysis. A task similarity index is obtained to characterize the difficulty of distinguishing marine debris categories; t∈[1,N-1]; based on the task similarity index, it is determined whether to enable RoI fine-grained discrimination modulation during the training of the current (t+1)th class incremental learning task; if yes, the RoI features in the target detection model training process are weighted and modulated based on an attention mechanism and then the original network structure is continued; if no, the original network structure of the target detection model is maintained for training; the above detection module is used to collect the target detection data to be identified by UAV, input it into the trained target detection model, and output the target detection results; the specific processing of the above functional modules can be referred to the detailed description in the above marine debris fine-grained identification method, which will not be repeated here.
[0034] The above description is merely a detailed explanation of preferred embodiments and principles of the present invention. For those skilled in the art, there may be changes in specific implementation methods based on the ideas provided by the present invention, and these changes should also be considered within the scope of protection of the present invention.
Claims
1. A method for fine-grained identification of marine debris based on UAV imagery, characterized in that, Includes the following steps: S1. Construct a target detection dataset based on data collected by UAVs and divide it into N incremental learning tasks. The target detection dataset contains different categories of marine debris, where N is an integer greater than 1. S2. Sequentially input the incremental learning tasks into the target detection model for training, obtaining the trained target detection model. During the training of the (t+1)th incremental learning task, samples are extracted from the data corresponding to the historical (t)th incremental learning tasks and the data of the current (t+1)th incremental learning task to form a prior analysis dataset. This prior analysis dataset is input into a prior analyzer for similarity analysis. , obtain the task similarity index used to characterize the difficulty of distinguishing marine debris categories; t∈[1,N-1]; based on the task similarity index, determine whether to enable RoI fine-grained discrimination modulation during the training of the current (t+1)th class incremental learning task; if yes, then perform weighted modulation of the RoI features in the target detection model training process based on the attention mechanism and continue processing of the original network structure; if no, then maintain the original network structure of the target detection model for training; S3, collect the target detection data to be identified through the UAV, input it into the trained target detection model, and output the target detection results.
2. The method for fine-grained identification of marine debris according to claim 1, characterized in that, Step S1 specifically includes the following steps: S11, collecting a target detection dataset containing multiple marine debris categories based on UAVs; wherein each data sample includes at least an input image and the corresponding target bounding box and category label; S12, dividing the target detection dataset into multiple non-overlapping category subsets according to all marine debris categories; S13, each category subset corresponds to a class incremental learning task, and is sequentially input into the target detection model for training in a predetermined order.
3. The method for fine-grained identification of marine debris according to claim 1, characterized in that, In step S2, the process of inputting the prior analysis dataset into the prior analyzer for similarity analysis includes the following steps: S21, inputting the prior analysis dataset into the target detection model that has completed the training of the t-th class incremental learning task, and extracting the feature vector corresponding to each target instance from the regional interest feature layer of the model; S22, aggregating the feature vectors of target instances belonging to the same marine debris category to obtain category feature prototypes; S23, calculating the intra-task similarity between the category feature prototypes in the current (t+1)-th class incremental learning task; calculating the cross-task similarity between the category feature prototypes in the current (t+1)-th class incremental learning task and the category feature prototypes in the historical t-th class incremental learning tasks; S24, accumulating the intra-task similarity and cross-task similarity and normalizing them to obtain the task similarity index.
4. The method for fine-grained identification of marine debris according to claim 3, characterized in that, The task similarity index of the (t+1)th class incremental learning task for: ;in, This represents the sum of similarities within each task. It is the sum of cross-task similarity. This represents the total number of category feature prototype pairs involved in the similarity calculation.
5. The method for fine-grained identification of marine debris according to claim 4, characterized in that, In step S2, determining whether RoI fine-grained discriminative modulation needs to be enabled during the training of the (t+1)th class incremental learning task based on the task similarity index includes the following steps: S25, determine whether t is 1; if yes, the current class incremental learning task is the second class incremental learning task, and proceed to step S26; if no, proceed to step S27; S26, determine whether the task similarity index of the second class incremental learning task is greater than the task similarity index of the first class incremental learning task; if yes, enable RoI fine-grained discriminative modulation; if no, do not enable RoI fine-grained discriminative modulation; S27, calculate the distance between the task similarity index of the current class incremental learning task and the task similarity index of the historical t class incremental learning tasks, determine the target historical class incremental learning task that is closest to the current class incremental learning task; and determine whether RoI fine-grained discriminative modulation is enabled during the training of the current class incremental learning task based on the enabled status of RoI fine-grained discriminative modulation of the target historical class incremental learning task during the training process.
6. The method for fine-grained identification of marine debris according to claim 5, characterized in that, The task similarity index of the current incremental learning task is compared with the historical ones. The distance between the task similarity metrics of each class of incremental learning tasks is: ;in, 。 7. The method for fine-grained identification of marine debris according to claim 6, characterized in that, The target historical incremental learning task is the first... Incremental learning task for each class: 。 8. The method for fine-grained identification of marine debris according to any one of claims 1-7, characterized in that, The attention-based weighted modulation process includes: first, global pooling of RoI features to obtain channel statistical vectors that reflect the strength of each channel response; then, setting attention branches for different types of incremental learning tasks during the activation phase; generating channel weight coefficients based on the channel statistical vectors by the attention branches of the corresponding tasks; and weighting the RoI features in a residual manner.
9. The method for fine-grained identification of marine debris according to any one of claims 1-7, characterized in that, The object detection model is the object detection framework Faster R-CNN.
10. A marine debris fine-grained identification system based on UAV imagery, employing the marine debris fine-grained identification method as described in any one of claims 1-9, characterized in that, The marine debris fine-grained identification system includes: a construction module for constructing a target detection dataset based on UAV data and dividing it into N class incremental learning tasks; wherein the target detection dataset contains different marine debris categories, and N is an integer greater than 1; and a training module for sequentially inputting the class incremental learning tasks into the target detection model for training to obtain the trained target detection model; wherein, during the training of the current (t+1)th class incremental learning task, samples are extracted from the data corresponding to the historical (t)th class incremental learning tasks and the current (t+1)th class incremental learning task to form a prior analysis dataset, and the prior analysis dataset is input into the prior analysis... The analyzer performs similarity analysis to obtain a task similarity index that characterizes the difficulty of distinguishing marine debris categories; t∈[1,N-1]; based on the task similarity index, it is determined whether to enable RoI fine-grained discrimination modulation during the training of the current (t+1)th class incremental learning task; if yes, the RoI features in the target detection model training process are weighted and modulated based on an attention mechanism and then the original network structure is continued; if no, the original network structure of the target detection model is maintained for training; the detection module is used to collect target detection data to be identified through UAV, input it into the trained target detection model, and output the target detection results.
Citation Information
Patent Citations
Continuous few-sample target detection method based on category registration mechanism and regional contrast learning
CN117292112A
Image classification method and system based on CLIP category incremental learning
CN118506049A
Image classification method based on incremental learning of multi-branch network architecture
CN119942234A
Ship category identification method based on incremental learning
CN120689852A
Small sample target detection method based on class knowledge constraint-self-adaption
CN120707827A