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

Star History