Skip to content

Latest commit

 

History

History
48 lines (34 loc) · 1.2 KB

ProposedModels.md

File metadata and controls

48 lines (34 loc) · 1.2 KB

Proposed Models for the Model Zoo

Slated Models

  1. EfficientNet-lite https://github.com/onnx/tensorflow-onnx/tree/master/tutorials

  2. EfficientNet-det + EfficientNet-edge (Waiting for perf to improve) https://github.com/onnx/tensorflow-onnx/tree/master/tutorials

  3. SSD-MobileNet v2 https://github.com/onnx/tensorflow-onnx/tree/master/tutorials

  4. YOLO v4

  5. BERT variations

  • PyTorch BERT export
  • RoBERTa
  • mobile-BERT
  • mobile-albert
  1. Other TF models
  • MobileNetEdgeTPU
  • DeepLab

  1. Deep Voice 3

  2. Visual Q & A - Bilinear Attention Networks https://github.com/jnhwkim/ban-vqa

Suggestions from Svetlana Levitan, IBM

  1. Audio Classifier https://github.com/IBM/MAX-Audio-Classifier

  2. Toxic Comment Classifier Detect 6 types of toxicity in user comments https://developer.ibm.com/exchanges/models/all/max-toxic-comment-classifier/

  3. Text Sentiment Classifier Detect the sentiment captured in short pieces of text https://developer.ibm.com/exchanges/models/all/max-text-sentiment-classifier/

  4. Speech to Text Converter Converts spoken words into text form. https://developer.ibm.com/technologies/artificial-intelligence/models/max-speech-to-text-converter/