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
- GhostNet
ObjectDetection-network
- SSD:
- YOLO:
- YOLOv2:
- YOLOv3:
- FCOS:
- FPN:
- RetinaNet
- Objects as Points:
- FSAF:
- CenterNet
- FoveaBox
Semantic Segmentation
-
FCN
-
Fast-SCNN
-
LEDNet:
-
LRNNet
-
FisheyeMODNet:
Instance Segmentation
- PolarMask
FaceDetectorAndRecognition
- FaceBoxes
- LFFD
- VarGFaceNet
HumanPoseEstimation
- Stacked Hourglass Networks
- Simple Baselines
- LPN