/mobilenetv2

MobileNetV2 Image Classfication Model implemented in TensorFlow 2.0 and PyTorch

Primary LanguagePythonMIT LicenseMIT

MobileNetV2

MobileNetV2 Image Classfication Model arXiv:1801.04381 [cs.CV] implementation in most popular frameworks

TensorFlow 2.0

TensorFlow model has a data pipeline that supports splitting data set into N folds allowing training N models.

PyTorch