search

Syn-McJ / TFClassify-Unity-Barracuda

An example of using Tensorflow and ONNX models with Unity Barracuda inference engine for image classification and object detection.
MIT License
124 stars 32 forks source link

barracuda not success to convert the pb file #2

Closed HolyShitKillingMachine closed 4 years ago

HolyShitKillingMachine commented 4 years ago

I am downing the ml-agents file from the github link you provide,and use anaconda prompt to cd to the destination of converter.py,and type the code ,python tensorflow_to_barracuda.py my.pb destination the output destination,And it cames out that there is no mlagents module. I donot know why

Syn-McJ commented 4 years ago

@HolyShitKillingMachine, You need to download the whole repo to use this script. Alternatively, you can try to install mlagents from pip like pip install mlagents.

Pouyan97 commented 4 years ago

@HolyShitKillingMachine Easier way to transform tensorflow to barracuda is by just downloading the whole repo in this link. Then move your tensorflow model 'my.pb' to barracuda-release/Tools folder. cd barracuda-release/Tools to get to the Tools folder and then just run python tensorflow_to_barracuda.py my.pb output.*