Figure 1.
Figure 2.
Figure 3.
Figure 4.
Figure 5.
Figure 6.
Figure 7.
Environment configuration table
| Hardware | Processor Video Cards | Intel(R)Core(TM) i7-6500U GeForce_RTX_2080_Ti |
|---|---|---|
| Software | Operating System | windows10 |
| Deep Learning Framework | pytorch-gpu | |
| Compiler Language | python | |
| Compilers | pycharm |
Default dimension and quantity of feature map
| Feature Map | Width and height of the feature map | Default boxes size | Number of default boxes |
| Feature Map1 | 38 × 38 | 21{1/2,1,2};
| 38 × 38 × 4 |
| Feature Map2 | 19 × 19 | 45{1/3,1/2,1,2,3};
| 19 × 19 × 6 |
| Feature Map3 | 10 × 10 | 99{1/3,1/2,1,2,3};
| 10 × 10 × 6 |
| Feature Map4 | 5 × 5 | 153{1/3,1/2,1,2,3};
| 5 × 5 × 6 |
| Feature Map5 | 3 × 3 | 207{1/2,1,2};
| 3 × 3 × 4 |
| Feature Map6 | 1 × 1 | 261{1/2,1,2};
| 1 × 1 × 4 |
Comparison of detection results of different algorithms on the PASCAL VOC2012 dataset
| Algorithm | mAP |
|---|---|
| SSD-VGG16 | 70.6% |
| SSD-ResNet50 | 72.1% |
| The improved algorithm | 72.7% |
Training parameters setting table
| Parameter | Value |
|---|---|
| learning rate | 0.0005 |
| momentum | 0.9 |
| weight_decay | 0.0005 |
| batch size | 16 |
| epoch | 50 |
| step_size | 5 |