Search Unity

  1. Looking for a job or to hire someone for a project? Check out the re-opened job forums.
    Dismiss Notice
  2. Unity 2020 LTS & Unity 2021.1 have been released.
    Dismiss Notice
  3. Good news ✨ We have more Unite Now videos available for you to watch on-demand! Come check them out and ask our experts any questions!
    Dismiss Notice

Image recognition

Discussion in 'ML-Agents' started by XSquare2, Apr 4, 2021.

  1. XSquare2

    XSquare2

    Joined:
    May 7, 2018
    Posts:
    6
    Hello, please help. Using MLAgents, I try to tune the neural network to determine the pictures. I create pictures of 30 pieces of the same suit and save them to a .bin file (an array of bool of size 1024 is saved, where the filled pixel is true) and feed this base to the neural network, types of picture 22. The neural network is trained, but it does not accurately determine the pictures, makes mistakes and confuses pictures ... Here is the code https://pastebin.com/ubx5mqAr. Am I doing everything right? And yet, if the neural network has not determined the pattern, then what to expect on the output neurons? or in general, how do you know that the neural network has not detected the drawing? Sorry for my English.
     
  2. ruoping_unity

    ruoping_unity

    Unity Technologies

    Joined:
    Jul 10, 2020
    Posts:
    77
    It looks like you're trying to tackle an image recognition problem.
    This kind of problem can be solved pretty well by supervised learning in general.
    ML-Agents is a toolkit that uses Reinforcement Learning to train behaviors for game agent so it might not be the best tool for your case. You might want to look for general image recognitions tools.

    Another point is you're feeding an image input as a vector and it will lose the information of relative position between each pixel and will affect the training performance.
     
  3. XSquare2

    XSquare2

    Joined:
    May 7, 2018
    Posts:
    6
    Understood thanks. So I need to use the CameraSensor and supervised learning?
     
unityunity