Skip to content

faizwhb/transfer_learning_toolkit

Repository files navigation

Transfer Learning Toolkit for Large Scale Image Classification

  • Image Classification Architectures:
    • DenseNets
    • Mobilenet - V2
    • ResNets
    • InceptionV3
  • Loss Functions
    • Categorical Cross Entropy
    • Categorical Cross Entropy with Label Smoothing and Temperature Scaling
  • Visualization
    • Tensorboard Summaries
    • Evaluation Metrics(Precision/Recall)
  • Optimizers
    • Adam
    • SWA with CLR
    • Budget Training
  • Datasets
    • Logo Classification
    • Flowers Classification

About

This repo contains some models and training utils for successfully training image classification models

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages