search

Unity-Technologies / ml-agents

The Unity Machine Learning Agents Toolkit (ML-Agents) is an open-source project that enables games and simulations to serve as environments for training intelligent agents using deep reinforcement learning and imitation learning.
https://unity.com/products/machine-learning-agents
Other
16.93k stars 4.14k forks source link

Barracuda IndexOutOfRangeException: Index was outside the bounds of the array. Error when I use Tensorflow Model #2143

Closed praptobala closed 5 years ago

praptobala commented 5 years ago

I am using Pretrained tensorflow CPM model for pose detection, the model is loading properly and showing the names of input/output layers. But crashing with "IndexOutOfRangeException: Index was outside the bounds of the array" error at the inference API.

Please let me know what would be cause, I found its some upsampling layer issue. But I cant change the model now. Any advice to fix the issue witout tweaking the model?

awjuliani commented 5 years ago

Hi @Praptobala

We currently do not officially support models not trained using our mlagents-learn package. That being said, we are looking to expand support to more model types in the future.

@mantasp May also be able to help, as he is one of the lead developers of the inference API.

praptobala commented 5 years ago

Thanks @awjuliani !

@mantasp Could you please help me on this issue? I can even send you the model file to look upon.

Thanks, Bala

mantasp commented 5 years ago

@Praptobala could you please share model (both TF and nn) + sample input? Thanks

praptobala commented 5 years ago

Hi @mantasp, sorry for late response.

Here's the link of .tflite model: https://drive.google.com/open?id=19mG6NgdJXUTyO7FWBB4Uu9IHYq1Y82Ua

Sample input is WebCam feed as scaled down input [1,192,192,3] + output [1, 96, 96, 28]

github-actions[bot] commented 3 years ago

This thread has been automatically locked since there has not been any recent activity after it was closed. Please open a new issue for related bugs.