Skip to content

Commit 40f102e

Browse files
author
Ruo-Ping (Rachel) Dong
authored
Add useVectorFrozenFlag option in FoodCollector (#4552)
1 parent 5131a66 commit 40f102e

File tree

2 files changed

+17
-5
lines changed

2 files changed

+17
-5
lines changed

Project/Assets/ML-Agents/Examples/FoodCollector/Prefabs/VisualFoodCollectorArea.prefab

Lines changed: 8 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -663,7 +663,7 @@ MonoBehaviour:
663663
myLaser: {fileID: 1900094563283840}
664664
contribute: 0
665665
useVectorObs: 0
666-
useFrozenFlag: 1
666+
useVectorFrozenFlag: 1
667667
--- !u!114 &114326390494230518
668668
MonoBehaviour:
669669
m_ObjectHideFlags: 0
@@ -681,6 +681,7 @@ MonoBehaviour:
681681
m_Width: 84
682682
m_Height: 84
683683
m_Grayscale: 0
684+
m_ObservationStacks: 1
684685
m_Compression: 1
685686
--- !u!114 &4034342608499629224
686687
MonoBehaviour:
@@ -1246,7 +1247,7 @@ MonoBehaviour:
12461247
myLaser: {fileID: 1307818939507544}
12471248
contribute: 0
12481249
useVectorObs: 0
1249-
useFrozenFlag: 1
1250+
useVectorFrozenFlag: 1
12501251
--- !u!114 &114429222608880102
12511252
MonoBehaviour:
12521253
m_ObjectHideFlags: 0
@@ -1264,6 +1265,7 @@ MonoBehaviour:
12641265
m_Width: 84
12651266
m_Height: 84
12661267
m_Grayscale: 0
1268+
m_ObservationStacks: 1
12671269
m_Compression: 1
12681270
--- !u!114 &7234640249101665162
12691271
MonoBehaviour:
@@ -1643,7 +1645,7 @@ MonoBehaviour:
16431645
myLaser: {fileID: 1898252046043334}
16441646
contribute: 0
16451647
useVectorObs: 0
1646-
useFrozenFlag: 1
1648+
useVectorFrozenFlag: 1
16471649
--- !u!114 &114036270357198286
16481650
MonoBehaviour:
16491651
m_ObjectHideFlags: 0
@@ -1661,6 +1663,7 @@ MonoBehaviour:
16611663
m_Width: 84
16621664
m_Height: 84
16631665
m_Grayscale: 0
1666+
m_ObservationStacks: 1
16641667
m_Compression: 1
16651668
--- !u!114 &3164735207755090463
16661669
MonoBehaviour:
@@ -3460,7 +3463,7 @@ MonoBehaviour:
34603463
myLaser: {fileID: 1779831409734062}
34613464
contribute: 0
34623465
useVectorObs: 0
3463-
useFrozenFlag: 1
3466+
useVectorFrozenFlag: 1
34643467
--- !u!114 &114322691115031348
34653468
MonoBehaviour:
34663469
m_ObjectHideFlags: 0
@@ -3478,6 +3481,7 @@ MonoBehaviour:
34783481
m_Width: 84
34793482
m_Height: 84
34803483
m_Grayscale: 0
3484+
m_ObservationStacks: 1
34813485
m_Compression: 1
34823486
--- !u!114 &5903164052970896384
34833487
MonoBehaviour:

Project/Assets/ML-Agents/Examples/FoodCollector/Scripts/FoodCollectorAgent.cs

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,6 +30,10 @@ public class FoodCollectorAgent : Agent
3030
public GameObject myLaser;
3131
public bool contribute;
3232
public bool useVectorObs;
33+
[Tooltip("Use only the frozen flag in vector observations. If \"Use Vector Obs\" " +
34+
"is checked, this option has no effect. This option is necessary for the " +
35+
"VisualFoodCollector scene.")]
36+
public bool useVectorFrozenFlag;
3337

3438
EnvironmentParameters m_ResetParams;
3539

@@ -49,9 +53,13 @@ public override void CollectObservations(VectorSensor sensor)
4953
var localVelocity = transform.InverseTransformDirection(m_AgentRb.velocity);
5054
sensor.AddObservation(localVelocity.x);
5155
sensor.AddObservation(localVelocity.z);
56+
sensor.AddObservation(m_Frozen);
5257
sensor.AddObservation(m_Shoot);
5358
}
54-
sensor.AddObservation(m_Frozen);
59+
else if (useVectorFrozenFlag)
60+
{
61+
sensor.AddObservation(m_Frozen);
62+
}
5563
}
5664

5765
public Color32 ToColor(int hexVal)

0 commit comments

Comments
 (0)