Content-Length: 261516 | pFad | http://github.com/popCain/TFtoCoreML/#start-of-content

B1 GitHub - popCain/TFtoCoreML: Transform the object detection model trained on TensorFlow to iOS CoreML model type
Skip to content

Transform the object detection model trained on TensorFlow to iOS CoreML model type

Notifications You must be signed in to change notification settings

popCain/TFtoCoreML

Repository files navigation

TFtoCoreML

Transform the object detection model trained on TensorFlow to iOS CoreML model type used tfcoreml(reference) and coremltools.

Folder List

  • mlmodels_IOU0.4_Conf0.6
    Core ML models(iOS) transformed from the models that trained on tensorflow(threshold: IOU=0.4; Confidece=0.6)
  • MobileNetV1_SSD/MobileNetV2_SSDLite/MobileNetV3_Large_SSDLite/MobileNetV3_Small_SSDLite/MobileDet_SSD_CPU

Transform Process

The transform process from file list

Original Model trained on tensorflow contains:preprocess + mobilenet_ssd + anchor generate + NMS + predictor parts.

  1. check_modelNodes.py
    Load the frozen_inference_graph.pb,and print the name of nodes in each layer. Then get the input_node_name(mobilenet) and output_node_name(scores/boundingboxes)
  2. frozenToSimplePB.py
    Simplify the model(Strip unused subgraphs get simplified frozen graph-ssd_mobilenet.pb)
  3. simplePB_ToCoreml.py
    Main network tranceform(ssd_mobilenet.mlmodel)
  4. Decoder.py


  1. NMS.py
  2. pipelines.py

About

Transform the object detection model trained on TensorFlow to iOS CoreML model type

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages









ApplySandwichStrip

pFad - (p)hone/(F)rame/(a)nonymizer/(d)eclutterfier!      Saves Data!


--- a PPN by Garber Painting Akron. With Image Size Reduction included!

Fetched URL: http://github.com/popCain/TFtoCoreML/#start-of-content

Alternative Proxies:

Alternative Proxy

pFad Proxy

pFad v3 Proxy

pFad v4 Proxy