MODEL_ZOO_en.md 5.0 KB
Newer Older
qq_30618961's avatar
qq_30618961 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30
# Model Libraries and Baselines

## Test Environment

- Python 3.7
- PaddlePaddle Daily version
- CUDA 10.1
- cuDNN 7.5
- NCCL 2.4.8

## General Settings

- All models were trained and tested in the COCO17 dataset.
- Unless special instructions, all the ResNet backbone network using [ResNet-B](https://arxiv.org/pdf/1812.01187) structure.
- **Inference time (FPS)**: The reasoning time was calculated on a Tesla V100 GPU by `tools/eval.py` testing all validation sets in FPS (number of pictures/second). CuDNN version is 7.5, including data loading, network forward execution and post-processing, and Batch size is 1.

## Training strategy

- We adopt and [Detectron](https://github.com/facebookresearch/Detectron/blob/master/MODEL_ZOO.md#training-schedules) in the same training strategy.
- 1x strategy indicates that when the total batch size is 8, the initial learning rate is 0.01, and the learning rate decreases by 10 times after 8 epoch and 11 epoch, respectively, and the final training is 12 epoch.
- 2X strategy is twice as much as strategy 1X, and the learning rate adjustment position is twice as much as strategy 1X.

## ImageNet pretraining model
Paddle provides a skeleton network pretraining model based on ImageNet. All pre-training models were trained by standard Imagenet 1K dataset. Res Net and Mobile Net are high-precision pre-training models obtained by cosine learning rate adjustment strategy or SSLD knowledge distillation training. Model details are available at [PaddleClas](https://github.com/PaddlePaddle/PaddleClas).


## Baseline

### Faster R-CNN

31
Please refer to[Faster R-CNN](https://github.com/PaddlePaddle/PaddleDetection/tree/release/2.5/configs/faster_rcnn/)
qq_30618961's avatar
qq_30618961 已提交
32 33 34

### Mask R-CNN

35
Please refer to[Mask R-CNN](https://github.com/PaddlePaddle/PaddleDetection/tree/release/2.5/configs/mask_rcnn/)
qq_30618961's avatar
qq_30618961 已提交
36 37 38

### Cascade R-CNN

39
Please refer to[Cascade R-CNN](https://github.com/PaddlePaddle/PaddleDetection/tree/release/2.5/configs/cascade_rcnn)
qq_30618961's avatar
qq_30618961 已提交
40 41 42

### YOLOv3

43
Please refer to[YOLOv3](https://github.com/PaddlePaddle/PaddleDetection/tree/release/2.5/configs/yolov3/)
qq_30618961's avatar
qq_30618961 已提交
44 45 46

### SSD

47
Please refer to[SSD](https://github.com/PaddlePaddle/PaddleDetection/tree/release/2.5/configs/ssd/)
qq_30618961's avatar
qq_30618961 已提交
48 49 50

### FCOS

51
Please refer to[FCOS](https://github.com/PaddlePaddle/PaddleDetection/tree/release/2.5/configs/fcos/)
qq_30618961's avatar
qq_30618961 已提交
52 53 54

### SOLOv2

55
Please refer to[SOLOv2](https://github.com/PaddlePaddle/PaddleDetection/tree/release/2.5/configs/solov2/)
qq_30618961's avatar
qq_30618961 已提交
56 57 58

### PP-YOLO

59
Please refer to[PP-YOLO](https://github.com/PaddlePaddle/PaddleDetection/tree/release/2.5/configs/ppyolo/)
qq_30618961's avatar
qq_30618961 已提交
60 61 62

### TTFNet

63
请参考[TTFNet](https://github.com/PaddlePaddle/PaddleDetection/tree/release/2.5/configs/ttfnet/)
qq_30618961's avatar
qq_30618961 已提交
64 65 66

### Group Normalization

67
Please refer to[Group Normalization](https://github.com/PaddlePaddle/PaddleDetection/tree/release/2.5/configs/gn/)
qq_30618961's avatar
qq_30618961 已提交
68 69 70

### Deformable ConvNets v2

71
Please refer to[Deformable ConvNets v2](https://github.com/PaddlePaddle/PaddleDetection/tree/release/2.5/configs/dcn/)
qq_30618961's avatar
qq_30618961 已提交
72 73 74

### HRNets

75
Please refer to[HRNets](https://github.com/PaddlePaddle/PaddleDetection/tree/release/2.5/configs/hrnet/)
qq_30618961's avatar
qq_30618961 已提交
76 77 78

### Res2Net

79
Please refer to[Res2Net](https://github.com/PaddlePaddle/PaddleDetection/tree/release/2.5/configs/res2net/)
qq_30618961's avatar
qq_30618961 已提交
80 81 82

### GFL

83
Please refer to[GFL](https://github.com/PaddlePaddle/PaddleDetection/tree/release/2.5/configs/gfl)
qq_30618961's avatar
qq_30618961 已提交
84 85 86

### PicoDet

87
Please refer to[PicoDet](https://github.com/PaddlePaddle/PaddleDetection/tree/release/2.5/configs/picodet)
qq_30618961's avatar
qq_30618961 已提交
88

F
Feng Ni 已提交
89 90
### PP-YOLOE

91
Please refer to[PP-YOLOE](https://github.com/PaddlePaddle/PaddleDetection/tree/release/2.5/configs/ppyoloe)
F
Feng Ni 已提交
92 93 94

### YOLOX

95 96 97 98
Please refer to[YOLOX](https://github.com/PaddlePaddle/PaddleDetection/tree/develop/configs/yolox)

### YOLOv5

F
Feng Ni 已提交
99
Please refer to[YOLOv5](https://github.com/PaddlePaddle/PaddleYOLO/tree/develop/configs/yolov5)
100 101 102

### YOLOv6

F
Feng Ni 已提交
103
Please refer to[YOLOv6](https://github.com/PaddlePaddle/PaddleYOLO/tree/develop/configs/yolov6)
104 105 106

### YOLOv7

F
Feng Ni 已提交
107
Please refer to[YOLOv7](https://github.com/PaddlePaddle/PaddleYOLO/tree/develop/configs/yolov7)
F
Feng Ni 已提交
108

qq_30618961's avatar
qq_30618961 已提交
109 110 111 112 113

## Rotating frame detection

### S2ANet

114
Please refer to[S2ANet](https://github.com/PaddlePaddle/PaddleDetection/tree/release/2.5/configs/dota/)
W
wangguanzhong 已提交
115

F
Feng Ni 已提交
116

W
wangguanzhong 已提交
117 118 119 120
## KeyPoint Detection

### PP-TinyPose

121
Please refer to [PP-TinyPose](https://github.com/PaddlePaddle/PaddleDetection/tree/release/2.5/configs/keypoint/tiny_pose)
W
wangguanzhong 已提交
122 123 124

## HRNet

125
Please refer to [HRNet](https://github.com/PaddlePaddle/PaddleDetection/tree/release/2.5/configs/keypoint/hrnet)
W
wangguanzhong 已提交
126 127 128

## HigherHRNet

129
Please refer to [HigherHRNet](https://github.com/PaddlePaddle/PaddleDetection/tree/release/2.5/configs/keypoint/higherhrnet)
W
wangguanzhong 已提交
130

F
Feng Ni 已提交
131

W
wangguanzhong 已提交
132 133
## Multi-Object Tracking

F
Feng Ni 已提交
134
### DeepSORT
W
wangguanzhong 已提交
135

136
Please refer to [DeepSORT](https://github.com/PaddlePaddle/PaddleDetection/tree/release/2.5/configs/mot/deepsort)
W
wangguanzhong 已提交
137 138 139

### JDE

140
Please refer to [JDE](https://github.com/PaddlePaddle/PaddleDetection/tree/release/2.5/configs/mot/jde)
W
wangguanzhong 已提交
141

F
Feng Ni 已提交
142 143
### FairMOT

144
Please refer to [FairMOT](https://github.com/PaddlePaddle/PaddleDetection/tree/release/2.5/configs/mot/fairmot)
F
Feng Ni 已提交
145 146

### ByteTrack
W
wangguanzhong 已提交
147

148
Please refer to [ByteTrack](https://github.com/PaddlePaddle/PaddleDetection/tree/release/2.5/configs/mot/bytetrack)