Search Unity

Using external machine learning models in unity

Discussion in 'Scripting' started by izdi, Jan 20, 2018.

  1. izdi

    izdi

    Joined:
    Dec 21, 2017
    Posts:
    1
    So I have a trained model dump from Caffe2, that basically classifies images. The question is:

    • How can I use it in Unity to build for mobile devices? Is there quick solution?

    • I found that only Tensorflow models are currently supported by unity, are there any other choices ?

    As far as I understand I need models built for different platforms (ios, android) to be loaded on initialization. Does it mean that I have to make Caffe2 unity plugin that loads the model dump for specific platform?
     
    izdi, Jan 20, 2018
    #1

  2. scheidtventureestate

    scheidtventureestate

    Joined:
    Nov 28, 2017
    Posts:
    33
    @izdi did you get any further in your question? I have a caffe2 model that I'd like to incorporate into unity 2018.1
     
    scheidtventureestate, Apr 2, 2018
    #2