| Literature DB >> 35784879 |
Bin Hu1, Yang Liu2,3, Pengzhi Chu1, Minglei Tong4, Qingjie Kong5.
Abstract
Object detection technology has been widely used in medical field, such as detecting the images of blood cell to count the changes and distribution for assisting the diagnosis of diseases. However, detecting small objects is one of the most challenging and important problems especially in medical scenarios. Most of the objects in medical images are very small but influential. Improving the detection performance of small objects is a very meaningful topic for medical detection. Current researches mainly focus on the extraction of small object features and data augmentation for small object samples, all of these researches focus on extracting the feature space of small objects better. However, in the training process of a detection model, objects of different sizes are mixed together, which may interfere with each other and affect the performance of small object detection. In this paper, we propose a method called pixel level balancing (PLB), which takes into account the number of pixels contained in the detection box as an impact factor to characterize the size of the inspected objects, and uses this as an impact factor. The training loss of each object of different size is adjusted by a weight dynamically, so as to improve the accuracy of small object detection. Finally, through experiments, we demonstrate that the size of objects in object detection interfere with each other. So that we can improve the accuracy of small object detection through PLB operation. This method can perform well with blood cell detection in our experiments.Entities:
Keywords: blood cell detection; medical image detection; object detection; pixel level balance; small object
Year: 2022 PMID: 35784879 PMCID: PMC9249342 DOI: 10.3389/fphys.2022.911297
Source DB: PubMed Journal: Front Physiol ISSN: 1664-042X Impact factor: 4.755
FIGURE 1Overview of our approach design.
Training accuracy effect of PLB2C (IOU = 0.50:0.95) on PyTorch.
| MAP | AP_small | AP_ Medium | AP_large | |
|---|---|---|---|---|
| Original faster_rcnn | 48.9 | 18.3 | 39.7 | 56.3 |
| PLB2C (default predict box) | 46.6 (−4.70%) | 19.2 (+4.92%) | 38.8 (−2.27%) | 53.9 (−4.26%) |
| PLB2C (matched labeled box) | 47.7 (−2.45%) | 16.8 (−8.20%) | 38.3 (−3.53%) | 55.1 (−2.13%) |
Comparison of the PLB methods (IOU = 0.50:0.95) on PyTorch.
| PLB1C | PLB1B | PLB2C | PLB2B | MAP | AP_small | AP_medium | AP_large |
|---|---|---|---|---|---|---|---|
| Not used | - | - | - | 48.9 | 18.3 | 39.7 | 56.3 |
| ✓ | - | - | - | 49.3 | 19.3 (+5.46%) | 40.2 | 56.5 |
| - | ✓ | - | - | 49.2 | 19.1 (+4.37%) | 39.9 | 56.8 |
| - | - | ✓ | - | 46.6 | 19.2 (+4.92%) | 38.8 | 53.9 |
| - | - | - | ✓ | 48.7 | 19.0 (+3.83%) | 39.2 | 56.2 |
| ✓ | - | ✓ | 49.3 | 18.6 (+1.64%) | 39.9 | 56.7 | |
| ✓ | ✓ | - | - | 49.1 | 19.3 (+5.46%) | 39.9 | 56.5 |
FIGURE 2Accuracy variation of small, medium and large object in PLB1C.
FIGURE 3Accuracy variation of small, medium and large object in PLB2C.
FIGURE 4Blood cell detection example from the BCCD data set. The red tags denote different classifications of the detected objects.
Accuracy effect with PLB in BBCD Dataset (IOU = 0.50:0.95) on PyTorch.
| MAP | AP_small | AP_ Medium | AP_large | |
|---|---|---|---|---|
| Origin model | 63.1 | 47.8 | 64.9 | 50.2 |
| PLB1C | 63.6 | 49.7 (+3.97%) | 65.2 (+0.46%) | 49.6 (−1.20%) |
| PLB1B | 63.1 | 48.6 (+1.67%) | 64.0 (−1.40%) | 50.6 (+0.80%) |
| PLB1C + PLB1B | 63.7 | 49.0 (+2.51%) | 66.6 (+2.62%) | 50.2 (+0%) |
Accuracy effect with PLB in BBCD Dataset (IOU = 0.50:0.95) on MindSpore.
| MAP | AP_small | AP_ Medium | AP_large | |
|---|---|---|---|---|
| Origin model | 62.9 | 48.7 | 65.4 | 48.0 |
| PLB1C | 63.3 | 51.4 (+5.54%) | 65.6 (+0.31%) | 48.3 (+0.63%) |
| PLB1B | 63.1 | 49.6 (+1.85%) | 64.2 (−1.83%) | 50.3 (+4.79%) |
| PLB1C + PLB1B | 63.6 | 50.4 (+3.49%) | 66.6 (+1.83%) | 49.2 (+2.50%) |
FIGURE 5Accuracy variation of PLB1C combined with PLB1B within BCCD dataset.