您最多选择25个主题 主题必须以中文或者字母或数字开头,可以包含连字符 (-),并且长度不得超过35个字符

68 行
2.4 KiB

using System;
using Unity.Simulation;
using UnityEngine.Perception.GroundTruth;
namespace UnityEngine.Perception.Randomization.Scenarios
{
/// <summary>
/// Derive this class to configure perception data capture while coordinating a scenario
/// </summary>
/// <typeparam name="T">The type of scenario constants to serialize</typeparam>
public abstract class PerceptionScenario<T> : Scenario<T> where T : ScenarioConstants, new()
{
/// <summary>
/// The guid used to identify this scenario's Iteration Metric Definition
/// </summary>
const string k_ScenarioIterationMetricDefinitionId = "DB1B258E-D1D0-41B6-8751-16F601A2E230";
/// <summary>
/// The metric definition used to report the current scenario iteration
/// </summary>
MetricDefinition m_IterationMetricDefinition;
/// <inheritdoc/>
protected override void OnAwake()
{
m_IterationMetricDefinition = DatasetCapture.RegisterMetricDefinition(
"scenario_iteration", "Iteration information for dataset sequences",
Guid.Parse(k_ScenarioIterationMetricDefinitionId));
}
/// <inheritdoc/>
protected override void OnStart()
{
var randomSeedMetricDefinition = DatasetCapture.RegisterMetricDefinition(
"random-seed",
"The random seed used to initialize the random state of the simulation. Only triggered once per simulation.",
Guid.Parse("14adb394-46c0-47e8-a3f0-99e754483b76"));
DatasetCapture.ReportMetric(randomSeedMetricDefinition, new[] { genericConstants.randomSeed });
}
/// <inheritdoc/>
protected override void OnIterationStart()
{
DatasetCapture.StartNewSequence();
DatasetCapture.ReportMetric(m_IterationMetricDefinition, new[]
{
new IterationMetricData { iteration = currentIteration }
});
}
/// <inheritdoc/>
protected override void OnComplete()
{
DatasetCapture.ResetSimulation();
Manager.Instance.Shutdown();
Quit();
}
/// <summary>
/// Used to report a scenario iteration as a perception metric
/// </summary>
struct IterationMetricData
{
// ReSharper disable once NotAccessedField.Local
public int iteration;
}
}
}