/PyTorch-Networks

Pytorch implementation of cnn network

Primary LanguagePython

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