/deep_learning

Implemented the training and inference of several common deep learning model algorithms with tensorflow and pytorch.

Primary LanguagePythonMIT LicenseMIT

Deep Learning

Implemented the training and inference of several common deep learning model algorithms with tensorflow and pytorch.

Darknet

  • VOC to YOLOLabel

PyTorch

  • DenseNet
  • DenseNet121
  • DenseNet161
  • DenseNet169
  • DenseNet201
  • LeNet_5
  • MobileNet_v2
  • ResNet18
  • ResNet34
  • ResNet50
  • ResNet101
  • ResNet152
  • ShuffleNet_V2_X2_0
  • SqueezeNet1_1

TensorFlow

  • AlexNet_V2
  • Caltech_256
  • Inception_V3
  • LeNet_5
  • ResNet_V2
  • VGG16
  • VOC to TFRecord