projection test for gallery space w/ chair using squeezed & gnomic projection.
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 

97 lines
2.4 KiB

using System.Collections;
using System.Collections.Generic;
using UnityEngine;
using Unity.MLAgents;
using Unity.Barracuda;
using Unity.MLAgents.Policies;
using System;
using Unity.Barracuda.ONNX;
using System.IO;
using System.Linq;
public class ModelSwap : MonoBehaviour
{
// Models
public NNModel m_InitialModel;
// public Agent agent;
public List<NNModel> nnModelList;
[HideInInspector]
public Walker agent;
[HideInInspector] public int currentModel = 0;
[HideInInspector] public string m_currentModelName = string.Empty;
[HideInInspector] public string m_PastModel = string.Empty;
// Path to .onnx files
string m_relPath = string.Empty;
private void OnEnable()
{
Waypoint.SwapModelOnWaypointReached += SwapModelOnReachingWaypoint;
}
private void OnDisable()
{
Waypoint.SwapModelOnWaypointReached -= SwapModelOnReachingWaypoint;
}
private void Start()
{
agent = this.GetComponent<Walker>();
}
// Set Model by name
public void SwitchModel(string modelName, Agent inst)
{
FindModelByName(modelName);
inst.SetModel("c_Walker", nnModelList[currentModel]);
if (modelName.Equals("Walker"))
{
agent.m_SelectedBrain = Walker.Brain.Walker;
m_currentModelName = modelName;
}
else if (modelName.Equals("Stairs"))
{
agent.m_SelectedBrain = Walker.Brain.DMScrambler;
m_currentModelName = modelName;
}
else if (modelName.Equals("Climber"))
{
agent.m_SelectedBrain = Walker.Brain.Climber;
m_currentModelName = modelName;
}
else if (modelName.Equals("Getup"))
{
agent.m_SelectedBrain = Walker.Brain.Getup;
m_currentModelName = modelName;
}
Debug.Log("Current Model: " + nnModelList[currentModel].name);
}
private void FindModelByName(string modelName)
{
int i = 0;
foreach (NNModel element in nnModelList)
{
if (element.name.Equals(modelName))
{
currentModel = i;
return;
}
i++;
}
}
private void SwapModelOnReachingWaypoint(string modelName)
{
SwitchModel(modelName, agent);
}
public void SetInitialModel()
{
SwitchModel(m_InitialModel.name, agent);
}
};