GitHub
5 年前
当前提交
734bad58
共有 5 个文件被更改,包括 116 次插入 和 2 次删除
-
18com.unity.ml-agents/Runtime/Sensor/CameraSensor.cs
-
44com.unity.ml-agents/Tests/Editor/Sensor/CameraSensorComponentTest.cs
-
11com.unity.ml-agents/Tests/Editor/Sensor/CameraSensorComponentTest.cs.meta
-
34com.unity.ml-agents/Tests/Editor/Sensor/CameraSensorTest.cs
-
11com.unity.ml-agents/Tests/Editor/Sensor/CameraSensorTest.cs.meta
|
|||
using System; |
|||
using NUnit.Framework; |
|||
using UnityEngine; |
|||
using MLAgents.Sensors; |
|||
|
|||
namespace MLAgents.Tests |
|||
{ |
|||
|
|||
[TestFixture] |
|||
public class CameraSensorComponentTest |
|||
{ |
|||
[Test] |
|||
public void TestCameraSensorComponent() |
|||
{ |
|||
foreach (var grayscale in new[] { true, false }) |
|||
{ |
|||
foreach (SensorCompressionType compression in Enum.GetValues(typeof(SensorCompressionType))) |
|||
{ |
|||
var width = 24; |
|||
var height = 16; |
|||
var camera = Camera.main; |
|||
|
|||
var agentGameObj = new GameObject("agent"); |
|||
|
|||
var cameraComponent = agentGameObj.AddComponent<CameraSensorComponent>(); |
|||
cameraComponent.camera = camera; |
|||
cameraComponent.height = height; |
|||
cameraComponent.width = width; |
|||
cameraComponent.grayscale = grayscale; |
|||
cameraComponent.compression = compression; |
|||
|
|||
var expectedShape = new[] { height, width, grayscale ? 1 : 3 }; |
|||
Assert.AreEqual(expectedShape, cameraComponent.GetObservationShape()); |
|||
Assert.IsTrue(cameraComponent.IsVisual()); |
|||
Assert.IsFalse(cameraComponent.IsVector()); |
|||
|
|||
var sensor = cameraComponent.CreateSensor(); |
|||
Assert.AreEqual(expectedShape, sensor.GetObservationShape()); |
|||
Assert.AreEqual(typeof(CameraSensor), sensor.GetType()); |
|||
} |
|||
} |
|||
} |
|||
} |
|||
} |
|
|||
fileFormatVersion: 2 |
|||
guid: 4c0b188faef38407e82223854fc8eaf5 |
|||
MonoImporter: |
|||
externalObjects: {} |
|||
serializedVersion: 2 |
|||
defaultReferences: [] |
|||
executionOrder: 0 |
|||
icon: {instanceID: 0} |
|||
userData: |
|||
assetBundleName: |
|||
assetBundleVariant: |
|
|||
using System; |
|||
using NUnit.Framework; |
|||
using UnityEngine; |
|||
using MLAgents.Sensors; |
|||
|
|||
namespace MLAgents.Tests |
|||
{ |
|||
|
|||
[TestFixture] |
|||
public class CameraSensorTest |
|||
{ |
|||
[Test] |
|||
public void TestCameraSensor() |
|||
{ |
|||
foreach (var grayscale in new[] { true, false }) |
|||
{ |
|||
foreach (SensorCompressionType compression in Enum.GetValues(typeof(SensorCompressionType))) |
|||
{ |
|||
var width = 24; |
|||
var height = 16; |
|||
var camera = Camera.main; |
|||
var sensor = new CameraSensor(camera, width, height, grayscale, "TestCameraSensor", compression); |
|||
|
|||
var writeAdapter = new WriteAdapter(); |
|||
var obs = sensor.GetObservationProto(writeAdapter); |
|||
|
|||
Assert.AreEqual((int) compression, (int) obs.CompressionType); |
|||
var expectedShape = new[] { height, width, grayscale ? 1 : 3 }; |
|||
Assert.AreEqual(expectedShape, obs.Shape); |
|||
} |
|||
} |
|||
} |
|||
} |
|||
} |
|
|||
fileFormatVersion: 2 |
|||
guid: ccdfc5b4015c9465cb1e811375be971c |
|||
MonoImporter: |
|||
externalObjects: {} |
|||
serializedVersion: 2 |
|||
defaultReferences: [] |
|||
executionOrder: 0 |
|||
icon: {instanceID: 0} |
|||
userData: |
|||
assetBundleName: |
|||
assetBundleVariant: |
撰写
预览
正在加载...
取消
保存
Reference in new issue