PyTorch-Models
Pytorch implementation of cnn network
Classical network
-
AlexNet:
-
VGG:
-
ResNet:
-
InceptionV1:
-
InceptionV2 and InceptionV3:
-
InceptionV4 and Inception-ResNet:
Lightweight network
- MobileNets:
- MobileNetV2：
- MobileNetV3：
- ShuffleNet：
- ShuffleNet V2:
- SqueezeNet
- Xception
- MixNet
ObjectDetection-network
- SSD:
- YOLO:
- YOLOv2:
- YOLOv3:
- FCOS:
- FPN:
- RetinaNet
- Objects as Points:
- FSAF:
- CenterNet
- FoveaBox
Semantic Segmentation
-
FCN:
-
FisheyeMODNet:
Instance Segmentation
- PolarMask
FaceDetectorAndRecognition
-
FaceBoxes
-
LFFD
HumanPoseEstimation
- Stacked Hourglass Networks