Over a million developers have joined DZone.

Reusing Open-Source Models in AI Applications

DZone's Guide to

Reusing Open-Source Models in AI Applications

Developing models for AI is usually not an easy task. Let's take a look at how open source is helping to change that.

· AI Zone ·
Free Resource

Did you know that 50- 80% of your enterprise business processes can be automated with AssistEdge?  Identify processes, deploy bots and scale effortlessly with AssistEdge.

Building machine and deep learning models from scratch is often not trivial, not for developers and sometimes not even for data scientists. Fortunately, over the last few years, several models have been developed and shared that can be reused and sometimes extended. This allows developers adding AI to applications without having to be data scientists.

This article describes how to find open source AI models and introduces briefly ONNX, which allows conversions between different model formats.

In order to find models, there isn't the one place to go to. In my experience, the best way to find models is to rely on Google searches. That being said, there are a number of websites with a lot of interesting open-source models.

IBM Model Asset Exchange

The IBM Model Asset Exchange doesn't contain too many models yet, but I like the consumability for developers. As I blogged about, the models are put in Docker containers and predictions can be done via REST APIs. The models are published under open source licenses and the IP has been vetted. Some of the models can be re-trained easily with custom data.

TensorFlow Models

As the name implies the TensorFlow Models GitHub project is specialized on TensorFlow. This catalog has a big amount of models and many different types of models. Many of the models are high quality and well documented, for example, the visual recognition models. For some of the other models, I'm not sure whether they are actively maintained. Also, some of the models are not the trained models, but code to execute trainings of models.

Caffe2 Model Zoo

The Caffe2 Model Zoo is another website you might want to check out when looking for models. There is a bigger number of visual recognition models available for Caffe and Caffe2. As a developer without much data science background however, I've found it sometimes hard to understand what these models do and how to use them.

ONNX Model Zoo

The Open Neural Network eXchange Model Zoo is a collection of pre-trained, state-of-the-art models in the ONNX format. ONNX is an open format to represent deep learning models. ONNX was initiated by Facebook and Microsoft, IBM joined shortly after this.

There are several converters available to import ONNX models in frameworks like TensorFlow, CoreML and Caffe and vice versa converters to convert models from different deep learning frameworks into the ONNX format. This allows developers, for example, to convert models that cannot be run natively on mobile devices into formats that can be executed on mobile devices. Unfortunately, I haven't had any luck to get a sample working so far, but I'll keep trying as the different converters evolve.

This screenshot shows the currently available ONNX converters:

Consuming AI in byte sized applications is the best way to transform digitally. #BuiltOnAI, EdgeVerve’s business application, provides you with everything you need to plug & play AI into your enterprise.  Learn more.

open source ,models ,deep learning ,ibm ,machine learning ,artificial intelligence ,neural network ,tensorflow ,onnx ,caffe2

Published at DZone with permission of

Opinions expressed by DZone contributors are their own.

{{ parent.title || parent.header.title}}

{{ parent.tldr }}

{{ parent.urlSource.name }}