# PytorchDetection

**Repository Path**: jjzhk-ai/pytorch-detection

## Basic Information

- **Project Name**: PytorchDetection
- **Description**: 使用Pytorch实现SSD，RFB，FSSD以及Yolo系列. 会持续更新
- **Primary Language**: Python
- **License**: MPL-2.0
- **Default Branch**: main
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 0
- **Forks**: 1
- **Created**: 2021-10-27
- **Last Updated**: 2025-02-21

## Categories & Tags

**Categories**: Uncategorized

**Tags**: Python, AI, 图像识别

## README

# 1.代码完成度

|       | SSD | RFB | FSSD | YOLOV1 | YOLOV2 | YOLOV3 | YOLOV4 | YOLOV5 |
| ----- | --- | --- | ---- | ------ | ------ | ------ | ------ | ------ |
| train |  👍 |   👍 |  👍  |   👍   |   👍    |   👍   |   👎   |   👎    |
|  eval |  👍 |   👍 |  👍  |   👍   |   👍    |   👍   |   👎   |   👎    |
|  test |  👍 |   👍 |  👍  |   👍   |   👍    |   👍   |   👍   |   👎    |

# 2.SSD
$300 \times 300$
|            |  vgg16  |  resnet50  |  resnet152  |  darknet19  |  mobilenetv1  |
| ---------- | :-----: | :--------: | :---------: | :---------: | :-----------: |
|   VOC-mAP  |  76.5%  |    79.2%   |    73.3%    |    74.4%    |     72.9%     |
|  VOC-Image | ![avatar](result/ssd_voc_vgg16_300.jpg) | ![avatar](result/ssd_voc_resnet50_300.jpg) | ![avatar](result/ssd_voc_resnet152_300.jpg) | ![avatar](result/ssd_voc_darknet19_300.jpg) | ![avatar](result/ssd_voc_mobilenetv1_300.jpg) |
|  COCO-mAP  |    🐦    |   25.0%    |     🐦      |    20.7%    |     18.8%     |
| COCO-Image |    🐦    | ![avatar](result/ssd_coco_resnet50_300.jpg) | 🐦 | ![avatar](result/ssd_coco_darknet19_300.jpg) | ![avatar](result/ssd_coco_mobilenetv1_300.jpg) |

# 3.RFB
$300 \times 300$
|            |  vgg16  |  resnet50  |  resnet152  |  darknet19  |  mobilenetv1  |
| ---------- | :-----: | :--------: | :---------: | :---------: | :-----------: |
|   VOC-mAP  |  79.0%  |    🐦      |     🐦       |    76.2%    |     73.8%     |
|  VOC-Image | ![avatar](result/rfb_voc_vgg16_300.jpg) | 🐦 | 🐦 | ![avatar](result/rfb_voc_darknet19_300.jpg) | ![avatar](result/rfb_voc_mobilenetv1_300.jpg) |
|  COCO-mAP  |   🐦     |     🐦     |     🐦       |    22.4%    |     19.0%     |
| COCO-Image |    🐦    | 🐦 | 🐦 | ![avatar](result/rfb_coco_darknet19_300.jpg) | ![avatar](result/rfb_coco_mobilenetv1_300.jpg) |

# 4.FSSD
$300 \times 300$
|            |  vgg16  |  resnet50  |  resnet152  |  darknet19  |  mobilenetv1  |
| ---------- | :-----: | :--------: | :---------: | :---------: | :-----------: |
|   VOC-mAP  |  77.9%  |    74.0%   |    74.1%    |    78.2%    |     73.5%     |
|  VOC-Image | ![avatar](result/fssd_voc_vgg16_300.jpg) | ![avatar](result/fssd_voc_resnet50_300.jpg) | ![avatar](result/fssd_voc_resnet152_300.jpg) | ![avatar](result/fssd_voc_darknet19_300.jpg) | ![avatar](result/fssd_voc_mobilenetv1_300.jpg) |
|  COCO-mAP  |    🐦    |    26.6%   |     🐦      |    25.2%    |     22.8%     |
| COCO-Image | 🐦 | ![avatar](result/fssd_coco_resnet50_300.jpg) | 🐦 | ![avatar](result/fssd_coco_darknet19_300.jpg) | ![avatar](result/fssd_coco_mobilenetv1_300.jpg) |

# 5.YOLO
### YOLOv1
|            |  Resnet50-7 | Resnet50-14 | resnet-7 | resnet-14 | vgg16-7 | vgg16-14 |
| ---------- | :---------: | :---------: | :------: | :-------: | :-----: | :------: |
|   VOC-mAP  |    61.1%    |      🐦     |  58.4%   |     🐦     |    🐦   |     🐦    |
|  VOC-Image | ![avatar](result/yolov1_voc_resnet50_7.jpg) | 🐦 | ![avatar](result/yolov1_voc_resnet_7.jpg) | 🐦 | 🐦 | 🐦 |
|  COCO-mAP  | 🐦 | 🐦 | 🐦 | 🐦 | 🐦 | 🐦 |
| COCO-Image | 🐦 | 🐦 | 🐦 | 🐦 | 🐦 | 🐦 |

### YOLOv2
|            |  darknet19 | darknet19-tiny | resnet50 |
| ---------- | :--------: | :------------: | :------: |
|   VOC-mAP  |    🐦      |       🐦        |    🐦     |
|  VOC-Image | ![avatar](result/yolov2_voc_darknet19.jpg) | ![avatar](result/yolov2_voc_darknet_tiny.jpg) | 🐦 |
|  COCO-mAP  |    🐦      |       🐦        |    🐦     |
| COCO-Image | ![avatar](result/yolov2_coco_darknet19.jpg) |       🐦        |    🐦     |

### YOLOv3
|            |  darknet53 |    tiny    |   spp   |
| ---------- | :--------: | :--------: | :-----: |
|   VOC-mAP  |    🐦      |     🐦      |   🐦    |
|  VOC-Image | ![avatar](result/yolov3_voc_darknet53.jpg) | ![avatar](result/yolov3_voc_tiny.jpg) | 🐦 |
|  COCO-mAP  |    🐦      |       🐦        |    🐦     |
| COCO-Image | ![avatar](result/yolov3_coco_darknet53.jpg) | ![avatar](result/yolov3_coco_tiny.jpg) | ![avatar](result/yolov3_coco_spp.jpg) |

### YOLOv4
|            |  darknet | pacsp | pacsp-mish | pacsp-s | pacsp-s-mish | pacsp-x | pacsp-x-mish | paspp | tiny |
| ---------- | :------: | :---: | :--------: | :-----: | :----------: | :-----: | :----------: | :---: | :--: |
|   VOC-mAP  | 🐦 | 🐦 | 🐦 | 🐦 | 🐦 | 🐦 | 🐦 | 🐦 | 🐦 |
|  VOC-Image | 🐦 | 🐦 | 🐦 | 🐦 | 🐦 | 🐦 | 🐦 | 🐦 | 🐦 |
|  COCO-mAP  | 🐦 | 🐦 | 🐦 | 🐦 | 🐦 | 🐦 | 🐦 | 🐦 | 🐦 |
| COCO-Image | ![avatar](result/yolov4_coco_darknet.jpg) | ![avatar](result/yolov4_coco_pacsp.jpg) | ![avatar](result/yolov4_coco_pacsp_mish.jpg) | ![avatar](result/yolov4_coco_pacsp_s.jpg) | ![avatar](result/yolov4_coco_pacsp_s_mish.jpg) | ![avatar](result/yolov4_coco_pacsp_x.jpg) | ![avatar](result/yolov4_coco_pacsp_x_mish.jpg) | 🐦 | 🐦 |

### YOLOv5