You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Copy file name to clipboardExpand all lines: README.md
+4-1Lines changed: 4 additions & 1 deletion
Display the source diff
Display the rich diff
Original file line number
Diff line number
Diff line change
@@ -38,8 +38,9 @@ Following models are implemented, each one also has a readme inside.
38
38
|[shufflenet](./shufflenetv2)| ShuffleNetV2 with 0.5x output channels |
39
39
|[squeezenet](./squeezenet)| SqueezeNet 1.1 model |
40
40
|[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)|
42
42
|[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)|
43
44
|[retinaface](./retinaface)| resnet-50, weights from [biubug6/Pytorch_Retinaface](https://github.com/biubug6/Pytorch_Retinaface)|
44
45
45
46
## Tricky Operations
@@ -61,13 +62,15 @@ Some tricky operations encountered in these models, already solved, but might ha
61
62
|upsample| replaced by a deconvolution layer, see yolov3. |
62
63
|hsigmoid| hard sigmoid is implemented as a plugin, hsigmoid and hswish are used in mobilenetv3 |
63
64
|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 |
0 commit comments