Search Unity

  1. Unity support for visionOS is now available. Learn more in our blog post.
    Dismiss Notice

Simple way to visualize NN graph

Discussion in 'ML-Agents' started by dhaesedavid, Oct 16, 2020.

  1. dhaesedavid

    dhaesedavid

    Joined:
    Oct 16, 2020
    Posts:
    3
    Hi,

    I want to verify the input layer caused by a RayPerceptionSensorComponent3D as it is being fed to the neural network. I tried to follow the instructions in this post and this post, but tensorboard seems unable to read the files raw_graph_def.pb or frozen_graph_def.pb. After renaming either of them to saved_model.pb I managed to run the Python script from the above posts without errors, but tensorboard gives me:

    upload_2020-10-16_9-56-27.png

    Do I need to tell Unity to export the graphs, somehow?

    Many thanks
     
  2. celion_unity

    celion_unity

    Joined:
    Jun 12, 2019
    Posts:
    289
    Netron (https://lutzroeder.github.io/netron/) is a good tool to visualize the networks, and it supports both .onnx and .nn files.

    However, the observations from RayPerceptionSensorComponent3D will get concatenated with the other vector observations into one large vector, so you won't see anything specifically referring to the sensor.
     
    dhaesedavid likes this.
  3. dhaesedavid

    dhaesedavid

    Joined:
    Oct 16, 2020
    Posts:
    3