Skip to content

Commit 660fb6f

Browse files
authored
Update README.md
1 parent bb79a84 commit 660fb6f

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

README.md

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,8 +38,9 @@ Following models are implemented, each one also has a readme inside.
3838
|[shufflenet](./shufflenetv2)| ShuffleNetV2 with 0.5x output channels |
3939
|[squeezenet](./squeezenet)| SqueezeNet 1.1 model |
4040
|[vgg](./vgg)| VGG 11-layer model |
41-
|[yolov3](./yolov3)| darknet-53, weights from yolov3 authors |
41+
|[yolov3](./yolov3)| darknet-53, weights from yolov3 authors, pytorch implementation from [ayooshkathuria/pytorch-yolo-v3](https://github.com/ayooshkathuria/pytorch-yolo-v3) |
4242
|[yolov3-spp](./yolov3-spp)| darknet-53, weights from [ultralytics/yolov3](https://github.com/ultralytics/yolov3) |
43+
|[yolov4](./yolov4)| CSPDarknet53, weights from [AlexeyAB/darknet](https://github.com/AlexeyAB/darknet#pre-trained-models), pytorch implementation from [ultralytics/yolov3](https://github.com/ultralytics/yolov3) |
4344
|[retinaface](./retinaface)| resnet-50, weights from [biubug6/Pytorch_Retinaface](https://github.com/biubug6/Pytorch_Retinaface) |
4445

4546
## Tricky Operations
@@ -61,13 +62,15 @@ Some tricky operations encountered in these models, already solved, but might ha
6162
|upsample| replaced by a deconvolution layer, see yolov3. |
6263
|hsigmoid| hard sigmoid is implemented as a plugin, hsigmoid and hswish are used in mobilenetv3 |
6364
|retinaface output decode| implement a plugin to decode bbox, confidence and landmarks, see retinaface. |
65+
|mish| mish activation is implemented as a plugin, mish is used in yolov4 |
6466

6567
## Speed Benchmark
6668

6769
| Models | Device | BatchSize | Mode | Input Shape(HxW) | FPS |
6870
|-|-|:-:|:-:|:-:|:-:|
6971
| YOLOv3(darknet53) | Xavier | 1 | FP16 | 320x320 | 55 |
7072
| YOLOv3-spp(darknet53) | Xeon E5-2620/GTX1080 | 1 | FP32 | 256x416 | 94 |
73+
| YOLOv4(CSPDarknet53) | Xeon E5-2620/GTX1080 | 1 | FP32 | 256x416 | 67 |
7174
| RetinaFace(resnet50) | TX2 | 1 | FP16 | 384x640 | 15 |
7275
| RetinaFace(resnet50) | Xeon E5-2620/GTX1080 | 1 | FP32 | 928x1600 | 15 |
7376

0 commit comments

Comments
 (0)